diff --git a/src/spikeinterface/benchmark/benchmark_merging.py b/src/spikeinterface/benchmark/benchmark_merging.py index d1ec6a41db..f3c8fbb272 100644 --- a/src/spikeinterface/benchmark/benchmark_merging.py +++ b/src/spikeinterface/benchmark/benchmark_merging.py @@ -25,12 +25,11 @@ def __init__(self, recording, splitted_sorting, params, gt_sorting, splitted_cel self.result = {} def run(self, **job_kwargs): - sorting_analyzer = create_sorting_analyzer(self.gt_sorting, self.recording, format="memory", sparse=True, **job_kwargs) + sorting_analyzer = create_sorting_analyzer( + self.gt_sorting, self.recording, format="memory", sparse=True, **job_kwargs + ) merged_analyzer, self.result["merges"], self.result["outs"] = auto_merge_units( - sorting_analyzer, - extra_outputs=True, - **self.method_kwargs, - **job_kwargs + sorting_analyzer, extra_outputs=True, **self.method_kwargs, **job_kwargs ) self.result["sorting"] = merged_analyzer.sorting diff --git a/src/spikeinterface/curation/auto_merge.py b/src/spikeinterface/curation/auto_merge.py index 7d19f707fd..a7174a3bf8 100644 --- a/src/spikeinterface/curation/auto_merge.py +++ b/src/spikeinterface/curation/auto_merge.py @@ -343,7 +343,7 @@ def compute_merge_unit_groups( params["censored_period_ms"], ) outs["pairs_decreased_score"] = pairs_decreased_score - + ind1, ind2 = np.nonzero(pair_mask) print(step, len(ind1)) diff --git a/src/spikeinterface/sorters/internal/spyking_circus2.py b/src/spikeinterface/sorters/internal/spyking_circus2.py index df0a22cfef..e86650b795 100644 --- a/src/spikeinterface/sorters/internal/spyking_circus2.py +++ b/src/spikeinterface/sorters/internal/spyking_circus2.py @@ -344,10 +344,10 @@ def final_cleaning_circus( recording, sorting, templates, - similarity_kwargs = {"method": "l2", "support": "union", "max_lag_ms": 0.1}, - apply_merge_kwargs = {"merging_mode": "soft", "sparsity_overlap": 0.5, "censor_ms": 3.0}, - correlograms_kwargs = {}, - template_diff_thresh = np.arange(0.05, 0.25, 0.05) + similarity_kwargs={"method": "l2", "support": "union", "max_lag_ms": 0.1}, + apply_merge_kwargs={"merging_mode": "soft", "sparsity_overlap": 0.5, "censor_ms": 3.0}, + correlograms_kwargs={}, + template_diff_thresh=np.arange(0.05, 0.25, 0.05), ): from spikeinterface.sortingcomponents.tools import create_sorting_analyzer_with_existing_templates