fix misnamed variable error

talmolab · Aug 30, 2024 · 1680ab7 · 1680ab7
1 parent f8fdcec
commit 1680ab7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dreem/datasets/tracking_dataset.py b/dreem/datasets/tracking_dataset.py
@@ -131,7 +131,7 @@ def make_train_splits(self, splits: tuple[float]) -> None:
             dataset_size = len(dataset_to_split)
 
             val_inds, test_inds = train_test_split(
-                range(test_size),
+                range(dataset_size),
                 train_size=val_frac,
                 test_size=test_frac,
             )