Added get_random_spikes method to MockWaveformExtractor

SpikeInterface · Mar 6, 2024 · feff156 · feff156
1 parent b24e5fa
commit feff156
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/src/spikeinterface/core/analyzer_extension_core.py b/src/spikeinterface/core/analyzer_extension_core.py
@@ -319,7 +319,9 @@ 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."
+        assert (
+            self.sorting_analyzer.has_extension("random_spikes")
+        ), "compute templates requires the random_spikes extension. You can run WaveformExtractor.get_random_spikes()"
         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

diff --git a/src/spikeinterface/core/waveforms_extractor_backwards_compatibility.py b/src/spikeinterface/core/waveforms_extractor_backwards_compatibility.py
@@ -131,7 +131,10 @@ def delete_waveforms(self) -> None:
         self.sorting_analyzer.delete_extension("waveforms")
 
     def delete_extension(self, extension) -> None:
-        self.sorting_analyzer.delete_extension(extension)
+        self.sorting_analyzer.delete_extension()
+
+    def get_random_spikes(self) -> None:
+        self.sorting_analyzer.compute("random_spikes")
 
     @property
     def recording(self) -> BaseRecording: