Update fslite/tests/generate_big_tests.py

Co-authored-by: codiumai-pr-agent-pro[bot] <151058649+codiumai-pr-agent-pro[bot]@users.noreply.github.com>
bigbio · Sep 25, 2024 · 0e24e2c · 0e24e2c
1 parent 0ccd98d
commit 0e24e2c
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/fslite/tests/generate_big_tests.py b/fslite/tests/generate_big_tests.py
@@ -36,10 +36,11 @@ def generate_large_test_dataset():
             chunk_labels = labels[chunk_start:chunk_end]
 
             # Generate chunk of features
-            chunk_features = {
-                f"feature{i}": np.random.rand(chunk_end - chunk_start)
+            # Generate chunk of features
+            chunk_features = (
+                (f"feature{i}", np.random.rand(chunk_end - chunk_start))
                 for i in range(1, n_features + 1)
-            }
+            )
 
             # Create DataFrame chunk
             chunk_data = {"sample_id": chunk_sample_ids, "label": chunk_labels}