Skip to content

Commit 718b743

Browse files
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>
1 parent 5a91f14 commit 718b743

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

fslite/tests/generate_big_tests.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -36,10 +36,11 @@ def generate_large_test_dataset():
3636
chunk_labels = labels[chunk_start:chunk_end]
3737

3838
# Generate chunk of features
39-
# Generate chunk of features
40-
chunk_features = (
41-
(f"feature{i}", np.random.rand(chunk_end - chunk_start))
39+
rng = np.random.default_rng()
40+
chunk_features = {
41+
f"feature{i}": rng.random(chunk_end - chunk_start)
4242
for i in range(1, n_features + 1)
43+
}
4344
)
4445

4546
# Create DataFrame chunk

0 commit comments

Comments
 (0)