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>

Author: ypriverol

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}