diff --git a/src/spikeinterface/core/analyzer_extension_core.py b/src/spikeinterface/core/analyzer_extension_core.py index 268513dac8..108aceab7e 100644 --- a/src/spikeinterface/core/analyzer_extension_core.py +++ b/src/spikeinterface/core/analyzer_extension_core.py @@ -319,6 +319,7 @@ def _compute_and_append(self, operators): # spikes = self.sorting_analyzer.sorting.to_spike_vector() # some_spikes = spikes[self.sorting_analyzer.random_spikes_indices] + assert self.sorting_analyzer.has_extension("random_spikes"), "compute templates requires the random_spikes extension." some_spikes = self.sorting_analyzer.get_extension("random_spikes").some_spikes() for unit_index, unit_id in enumerate(unit_ids): spike_mask = some_spikes["unit_index"] == unit_index