Skip to content

Commit

Permalink
[pre-commit.ci] auto fixes from pre-commit.com hooks
Browse files Browse the repository at this point in the history
for more information, see https://pre-commit.ci
  • Loading branch information
pre-commit-ci[bot] committed Oct 29, 2024
1 parent 2f06bb6 commit 2bf61e9
Show file tree
Hide file tree
Showing 3 changed files with 9 additions and 10 deletions.
9 changes: 4 additions & 5 deletions src/spikeinterface/benchmark/benchmark_merging.py
Original file line number Diff line number Diff line change
Expand Up @@ -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

Expand Down
2 changes: 1 addition & 1 deletion src/spikeinterface/curation/auto_merge.py
Original file line number Diff line number Diff line change
Expand Up @@ -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))

Expand Down
8 changes: 4 additions & 4 deletions src/spikeinterface/sorters/internal/spyking_circus2.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit 2bf61e9

Please sign in to comment.