From 57dcac308349dfce71bb760fea44169ca37238fc Mon Sep 17 00:00:00 2001 From: Alessio Buccino Date: Fri, 10 May 2024 13:14:07 +0200 Subject: [PATCH] Update src/spikeinterface/curation/splitunitsorting.py --- src/spikeinterface/curation/splitunitsorting.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/spikeinterface/curation/splitunitsorting.py b/src/spikeinterface/curation/splitunitsorting.py index 83d3acb02b..5854d1b64a 100644 --- 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