rollback 2nd try

MannLabs · Jun 7, 2024 · 4b39315 · 4b39315
1 parent 4c1c16b
commit 4b39315
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/src/sparcscore/ml/datasets.py b/src/sparcscore/ml/datasets.py
@@ -341,8 +341,7 @@ def __init__(self,
 
         self.dir_list = dir_list 
         self.target_col = target_col
-        index_list = index_list.sort()
-        self.index_list = index_list
+        self.index_list = index_list.sort()
         self.hours = hours
         self.root_dir = root_dir 
         self.transform = transform