Update auxfun_data.py

KonradDanielewski · Jan 30, 2024 · 78041b0 · 78041b0
1 parent 12812b2
commit 78041b0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/seba/data/auxfun_data.py b/seba/data/auxfun_data.py
@@ -116,7 +116,7 @@ def responsive_neurons2events(data_folder: str | list, data_obj: dict):
     responsive_units = data_obj["responsive_units"]
 
     for folder, rat in zip(data_folder, responsive_units):
-        df = pd.read_csv(os.path.join(folder, "cluster_info_good.csv"), index_col='id')
+        df = pd.read_csv(os.path.join(folder, "cluster_info_good.csv"), index_col="cluster_id")
         col_names = responsive_units[rat].keys()
         for col_name in col_names:
             df[col_name] = np.nan
@@ -152,4 +152,4 @@ def responsive_units_wilcoxon(data_obj: dict, conditions: List, rats: str, pre_e
         place = update.iloc[i].name
         places = place.split("#")
         responsive_units[places[0]][places[1]].append(int(places[2]))
-    return responsive_units
+    return responsive_units