sort list 2nd try

MannLabs · Jun 7, 2024 · 4c1c16b · 4c1c16b
1 parent 8674c82
commit 4c1c16b
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/sparcscore/ml/datasets.py b/src/sparcscore/ml/datasets.py
@@ -341,7 +341,8 @@ def __init__(self,
 
         self.dir_list = dir_list 
         self.target_col = target_col
-        self.index_list = index_list.sort() # list of indices to select
+        index_list = index_list.sort()
+        self.index_list = index_list
         self.hours = hours
         self.root_dir = root_dir 
         self.transform = transform