diff --git a/tests/core/partitioning/partitioning_test.bzl b/tests/core/partitioning/partitioning_test.bzl
index 0998d73b90..4f0e7740f6 100644
--- a/tests/core/partitioning/partitioning_test.bzl
+++ b/tests/core/partitioning/partitioning_test.bzl
@@ -1,14 +1,14 @@
-def partitioning_test(name, visibility=None):
-    native.cc_test(
-        name = name,
-        srcs = [name + ".cpp"],
-        visibility = visibility,
-        deps = [
-            "//tests/util",
-            "@googletest//:gtest_main",
-        ] + select({
-            ":use_pre_cxx11_abi":  ["@libtorch_pre_cxx11_abi//:libtorch"],
-            "//conditions:default":  ["@libtorch//:libtorch"],
-        }),
-        timeout="short"
-    )
+def partitioning_test(name, visibility = None):
+    native.cc_test(
+        name = name,
+        srcs = [name + ".cpp"],
+        visibility = visibility,
+        deps = [
+            "//tests/util",
+            "@googletest//:gtest_main",
+        ] + select({
+            ":use_pre_cxx11_abi": ["@libtorch_pre_cxx11_abi//:libtorch"],
+            "//conditions:default": ["@libtorch//:libtorch"],
+        }),
+        timeout = "short",
+    )