From 4c1c16b66234e37cc5326d2ea7290605c335922c Mon Sep 17 00:00:00 2001
From: namsaraeva <a.namsaraeva@tum.de>
Date: Fri, 7 Jun 2024 14:36:14 +0200
Subject: [PATCH] sort list 2nd try

---
 src/sparcscore/ml/datasets.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/sparcscore/ml/datasets.py b/src/sparcscore/ml/datasets.py
index f2a69c98..b4f78a5a 100644
--- 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