Merge pull request #2067 from samuelgarcia/various_fix

waveform extactor reload
SpikeInterface · Oct 4, 2023 · af9660a · af9660a
2 parents ca48f6b + 0c97fc4
commit af9660a
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/src/spikeinterface/core/waveform_extractor.py b/src/spikeinterface/core/waveform_extractor.py
@@ -175,7 +175,12 @@ def load_from_folder(
             rec_attributes = None
 
         if sorting is None:
-            sorting = load_extractor(folder / "sorting.json", base_folder=folder)
+            if (folder / "sorting.json").exists():
+                sorting = load_extractor(folder / "sorting.json", base_folder=folder)
+            elif (folder / "sorting.pickle").exists():
+                sorting = load_extractor(folder / "sorting.pickle")
+            else:
+                raise FileNotFoundError("load_waveforms() impossible to find the sorting object (json or pickle)")
 
         # the sparsity is the sparsity of the saved/cached waveforms arrays
         sparsity_file = folder / "sparsity.json"