From 3879b9bce68c9a0f09e7c7940c65034065686497 Mon Sep 17 00:00:00 2001 From: Rishi Chandra Date: Thu, 26 Dec 2024 01:47:39 +0000 Subject: [PATCH] formatting --- python/tests/test_umap.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/python/tests/test_umap.py b/python/tests/test_umap.py index 5e53d6be..9f4f8cd2 100644 --- a/python/tests/test_umap.py +++ b/python/tests/test_umap.py @@ -510,7 +510,9 @@ def test_umap_chunking( input_raw_data = X.get() nbytes = input_raw_data.nbytes - umap = UMAP(num_workers=gpu_number, random_state=random_state).setFeaturesCol("features") + umap = UMAP(num_workers=gpu_number, random_state=random_state).setFeaturesCol( + "features" + ) assert umap.max_records_per_batch == int(maxRecordsPerBatch) assert nbytes > BROADCAST_LIMIT @@ -536,7 +538,7 @@ def test_umap_chunking( def test_umap_sample_fraction(gpu_number: int) -> None: n_rows = 5000 - n_cols = 10 + n_cols = 10 random_state = 42 sample_fraction = 0.5