From 0e24e2cedd39ec04c1ac076df3d47eae797d678f Mon Sep 17 00:00:00 2001 From: Yasset Perez-Riverol Date: Wed, 25 Sep 2024 13:13:08 +0100 Subject: [PATCH] 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> --- fslite/tests/generate_big_tests.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/fslite/tests/generate_big_tests.py b/fslite/tests/generate_big_tests.py index 68413dd..1ea7a31 100644 --- 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}