formatting

Signed-off-by: Erik Ordentlich <[email protected]>
NVIDIA · Dec 30, 2023 · 4868fa4 · 4868fa4
1 parent 7aae4c2
commit 4868fa4
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/python/tests/test_kmeans.py b/python/tests/test_kmeans.py
@@ -152,7 +152,11 @@ def test_kmeans_basic(gpu_number: int, tmp_path: str) -> None:
             .map(lambda row: (row,))
             .toDF(["features"])
         )
-        kmeans = KMeans(num_workers=gpu_number, n_clusters=2).setFeaturesCol("features").setSeed(0)
+        kmeans = (
+            KMeans(num_workers=gpu_number, n_clusters=2)
+            .setFeaturesCol("features")
+            .setSeed(0)
+        )
 
         def assert_kmeans_model(model: KMeansModel) -> None:
             assert len(model.cluster_centers_) == 2