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 · 718b743 · 718b743
1 parent 5a91f14
commit 718b743
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
-            # Generate chunk of features
-            chunk_features = (
-                (f"feature{i}", np.random.rand(chunk_end - chunk_start))
+            rng = np.random.default_rng()
+            chunk_features = {
+                f"feature{i}": rng.random(chunk_end - chunk_start)
                 for i in range(1, n_features + 1)
+            }
             )
 
             # Create DataFrame chunk