Merge pull request #2888 from zm711/test-fix-2887

Put get_extension (to trigger auto-import) before get_children_dependencies for deleting Analyzer extensions to prevent KeyError
SpikeInterface · May 21, 2024 · 35d0edd · 35d0edd
2 parents 024c24a + 55cbae2
commit 35d0edd
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/spikeinterface/core/sortinganalyzer.py b/src/spikeinterface/core/sortinganalyzer.py
@@ -921,11 +921,11 @@ def compute_one_extension(self, extension_name, save=True, **kwargs):
         >>> wfs = compute_waveforms(sorting_analyzer, **some_params)
 
         """
+        extension_class = get_extension_class(extension_name)
+
         for child in _get_children_dependencies(extension_name):
             self.delete_extension(child)
 
-        extension_class = get_extension_class(extension_name)
-
         if extension_class.need_job_kwargs:
             params, job_kwargs = split_job_kwargs(kwargs)
         else: