From 79836e3a8981a1101c41f84b716850e52d6a9845 Mon Sep 17 00:00:00 2001 From: Alessio Buccino Date: Tue, 29 Apr 2025 09:46:31 +0200 Subject: [PATCH 1/2] Skip loading extensions when instantiating benchmarks --- src/spikeinterface/benchmark/benchmark_base.py | 8 +------- 1 file changed, 1 insertion(+), 7 deletions(-) diff --git a/src/spikeinterface/benchmark/benchmark_base.py b/src/spikeinterface/benchmark/benchmark_base.py index ff639bff19..eb2487a8bb 100644 --- a/src/spikeinterface/benchmark/benchmark_base.py +++ b/src/spikeinterface/benchmark/benchmark_base.py @@ -176,18 +176,12 @@ def scan_folder(self): self.levels = self.info["levels"] for key, folder in self.analyzers_path.items(): - analyzer = load_sorting_analyzer(folder) + analyzer = load_sorting_analyzer(folder, load_extensions=False) self.analyzers[key] = analyzer # the sorting is in memory here we take the saved one because comparisons need to pickle it later sorting = load(analyzer.folder / "sorting") self.datasets[key] = analyzer.recording, sorting - # for rec_file in (self.folder / "datasets" / "recordings").glob("*.pickle"): - # key = rec_file.stem - # rec = load(rec_file) - # gt_sorting = load(self.folder / f"datasets" / "gt_sortings" / key) - # self.datasets[key] = (rec, gt_sorting) - with open(self.folder / "cases.pickle", "rb") as f: self.cases = pickle.load(f) From e70ecac4b2cf3af5eb400ba789cdeb9049e652dd Mon Sep 17 00:00:00 2001 From: Alessio Buccino Date: Tue, 29 Apr 2025 09:56:33 +0200 Subject: [PATCH 2/2] Fix colors in plot vs snr --- src/spikeinterface/benchmark/benchmark_plot_tools.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/spikeinterface/benchmark/benchmark_plot_tools.py b/src/spikeinterface/benchmark/benchmark_plot_tools.py index 4cdc500b36..53e251e462 100644 --- a/src/spikeinterface/benchmark/benchmark_plot_tools.py +++ b/src/spikeinterface/benchmark/benchmark_plot_tools.py @@ -494,7 +494,9 @@ def plot_performances_vs_snr( colors = study.get_colors(levels_to_group_by=levels_to_keep) - assert all([key in colors for key in case_keys]), f"colors must have a color for each case key: {case_keys}" + assert all( + [key in colors for key in case_group_keys] + ), f"colors must have a color for each case key: {case_group_keys}" for key, key_list in case_group_keys.items(): color = colors[key]