Update histology.py

For neuropixels 1 in herbs probe has no index in filename.
KonradDanielewski · Jan 17, 2024 · a161bb4 · a161bb4
1 parent 9d4f008
commit a161bb4
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/seba/herbs_histology/histology.py b/seba/herbs_histology/histology.py
@@ -76,7 +76,10 @@ def get_brain_regions(data_folder:str, histology_folder:str, neuropixels_20=True
         all_shanks = []
 
         for shank in shanks:
-            obj = pd.read_pickle(histology + f"\\probe {shank}.pkl")
+            if neuropixels_20:
+                obj = pd.read_pickle(histology + f"\\probe {shank}.pkl")
+            else:
+                obj = pd.read_pickle(histology + f"\\probe.pkl")
             keys = pd.Series(obj["data"]["region_label"]).unique()
             values = pd.Series(obj["data"]["label_name"]).unique()