Skip to content

Commit 7c99530

Browse files
authored
Merge pull request #3916 from samuelgarcia/fix_benchmark_load
fix benchmark load with partial result
2 parents 85f4ff3 + 9c38887 commit 7c99530

File tree

1 file changed

+10
-3
lines changed

1 file changed

+10
-3
lines changed

src/spikeinterface/benchmark/benchmark_base.py

Lines changed: 10 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -594,15 +594,22 @@ def load_folder(cls, folder):
594594
elif format == "sorting":
595595
from spikeinterface.core import load_extractor
596596

597-
result[k] = load(folder / k)
597+
sorting_folder = folder / k
598+
if sorting_folder.exists():
599+
result[k] = load(sorting_folder)
598600
elif format == "Motion":
599601
from spikeinterface.core.motion import Motion
600602

601-
result[k] = Motion.load(folder / k)
603+
motion_folder = folder / k
604+
if motion_folder.exists():
605+
result[k] = Motion.load(motion_folder)
602606
elif format == "zarr_templates":
603607
from spikeinterface.core.template import Templates
604608

605-
result[k] = Templates.from_zarr(folder / k)
609+
zarr_folder = folder / k
610+
if zarr_folder.exists():
611+
612+
result[k] = Templates.from_zarr(zarr_folder)
606613

607614
return result
608615

0 commit comments

Comments
 (0)