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 Nov 22, 2024
1 parent c5fecf4 commit 28431f4
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions src/spikeinterface/sorters/internal/spyking_circus2.py
Original file line number Diff line number Diff line change
Expand Up @@ -413,14 +413,15 @@ def divide_recording(recording, seed=None, **job_kwargs):
def final_cleaning_circus(recording, sorting, templates, merging_kwargs, **job_kwargs):

from spikeinterface.core.sorting_tools import apply_merges_to_sorting

sa = create_sorting_analyzer_with_templates(sorting, recording, templates)

sa.compute("unit_locations", method="monopolar_triangulation", **job_kwargs)
similarity_kwargs = merging_kwargs.pop("similarity_kwargs", {})
sa.compute("template_similarity", **similarity_kwargs, **job_kwargs)
correlograms_kwargs = merging_kwargs.pop("correlograms_kwargs", {})
sa.compute("correlograms", **correlograms_kwargs, **job_kwargs)

auto_merge_kwargs = merging_kwargs.pop("auto_merge", {})
merges = get_potential_auto_merge(sa, resolve_graph=True, **auto_merge_kwargs)
sorting = apply_merges_to_sorting(sa.sorting, merges)
Expand Down

0 comments on commit 28431f4

Please sign in to comment.