fix(curatedclustering): correctly account for channel2electrode mapping

ttngu207 · Apr 8, 2024 · f723c67 · f723c67
1 parent 76e5834
commit f723c67
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/element_array_ephys/ephys_organoids.py b/element_array_ephys/ephys_organoids.py
@@ -763,7 +763,7 @@ def make(self, key):
         electrode_query &= f'electrode IN {tuple(probe_info["used_electrodes"])}'
 
         channel_info = electrode_query.fetch(as_dict=True, order_by="electrode")
-        channel_info: dict[int, dict] = {ch["electrode"]: ch for ch in channel_info}
+        channel_info: dict[int, dict] = {ch_idx: ch for ch_idx, ch in enumerate(channel_info)}
 
         # Get unit id to quality label mapping
         try: