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 Sep 28, 2023
1 parent daddd8c commit d7dcbe0
Showing 1 changed file with 1 addition and 2 deletions.
3 changes: 1 addition & 2 deletions src/spikeinterface/sortingcomponents/matching/circus.py
Original file line number Diff line number Diff line change
Expand Up @@ -128,7 +128,6 @@ def _freq_domain_conv(in1, in2, axes, shape, cache, calc_fast_len=True):
return ret



def compute_overlaps(templates, num_samples, num_channels, sparsities):
num_templates = len(templates)

Expand Down Expand Up @@ -475,6 +474,7 @@ def main_function(cls, traces, d):

return spikes


class CircusOMPSVDPeeler(BaseTemplateMatchingEngine):
"""
Orthogonal Matching Pursuit inspired from Spyking Circus sorter
Expand Down Expand Up @@ -637,7 +637,6 @@ def initialize_and_check_kwargs(cls, recording, kwargs):
d["unit_overlaps_tables"][i] = np.zeros(d["num_templates"], dtype=int)
d["unit_overlaps_tables"][i][d["unit_overlaps_indices"][i]] = np.arange(len(d["unit_overlaps_indices"][i]))


omp_min_sps = d["omp_min_sps"]
# d["stop_criteria"] = omp_min_sps * np.sqrt(d["noise_levels"].sum() * d["num_samples"])
d["stop_criteria"] = omp_min_sps * np.maximum(d["norms"], np.sqrt(d["noise_levels"].sum() * d["num_samples"]))
Expand Down

0 comments on commit d7dcbe0

Please sign in to comment.