Update src/spikeinterface/curation/splitunitsorting.py

SpikeInterface · May 10, 2024 · 57dcac3 · 57dcac3
1 parent fcb1c1a
commit 57dcac3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/spikeinterface/curation/splitunitsorting.py b/src/spikeinterface/curation/splitunitsorting.py
@@ -48,8 +48,8 @@ def __init__(self, parent_sorting, split_unit_id, indices_list, new_unit_ids=Non
         # make sure indices list is between 0 and tot_splits - 1
         indices_zero_based = [np.zeros_like(indices) for indices in indices_list]
         for segment_index in range(len(indices_list)):
-            for i, split_unit_idx in enumerate(split_unit_indices):
-                indices_zero_based[segment_index][indices_list[segment_index] == split_unit_idx] = i
+            for zero_based_index, split_unit_idx in enumerate(split_unit_indices):
+                indices_zero_based[segment_index][indices_list[segment_index] == split_unit_idx] = zero_based_index
 
         if new_unit_ids is None:
             # select new_unit_ids greater that the max id, event greater than the numerical str ids