Adding unit_ids

SpikeInterface · Sep 13, 2023 · dda7803 · dda7803
1 parent cc79213
commit dda7803
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/spikeinterface/sortingcomponents/clustering/random_projections.py b/src/spikeinterface/sortingcomponents/clustering/random_projections.py
@@ -177,7 +177,8 @@ def main_function(cls, recording, peaks, params):
             mode = "folder"
 
         sorting_folder = tmp_folder / "sorting"
-        sorting = NumpySorting.from_times_labels(spikes["sample_index"], spikes["unit_index"], fs)
+        unit_ids = np.arange(len(np.unique(spikes["unit_index"])))
+        sorting = NumpySorting(spikes, fs, unit_ids=unit_ids)
         sorting = sorting.save(folder=sorting_folder)
         we = extract_waveforms(
             recording,