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,
  )