Issue with kept indices

SpikeInterface · Jul 1, 2024 · 14d2002 · 14d2002
1 parent c37dae3
commit 14d2002
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/spikeinterface/core/analyzer_extension_core.py b/src/spikeinterface/core/analyzer_extension_core.py
@@ -84,7 +84,7 @@ def _merge_extension_data(
             valid = kept_indices[self.sorting_analyzer.get_extension("random_spikes")._get_data()]
             nb_skipped = np.cumsum(~kept_indices)
             new_data["random_spikes_indices"] = np.flatnonzero(valid)
-            new_data["random_spikes_indices"] -= nb_skipped[valid]
+            new_data["random_spikes_indices"] -= nb_skipped[new_data["random_spikes_indices"]]
         else:
             new_data["random_spikes_indices"] = self.data["random_spikes_indices"]
         return new_data