Skip to content

Commit c2ca6e9

Browse files
committed
fix benchmark load with partial result
1 parent 9dec50f commit c2ca6e9

1 file changed

Lines changed: 9 additions & 5 deletions

File tree

src/spikeinterface/benchmark/benchmark_base.py

Lines changed: 9 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -593,16 +593,20 @@ def load_folder(cls, folder):
593593
result[k] = pickle.load(f)
594594
elif format == "sorting":
595595
from spikeinterface.core import load_extractor
596-
597-
result[k] = load(folder / k)
596+
sorting_folder = folder / k
597+
if sorting_folder.exists():
598+
result[k] = load(sorting_folder)
598599
elif format == "Motion":
599600
from spikeinterface.core.motion import Motion
600-
601-
result[k] = Motion.load(folder / k)
601+
motion_folder = folder / k
602+
if motion_folder.exists():
603+
result[k] = Motion.load(motion_folder)
602604
elif format == "zarr_templates":
603605
from spikeinterface.core.template import Templates
606+
zarr_folder = folder / k
607+
if zarr_folder.exists():
604608

605-
result[k] = Templates.from_zarr(folder / k)
609+
result[k] = Templates.from_zarr(zarr_folder)
606610

607611
return result
608612

0 commit comments

Comments
 (0)