Skip to content

Commit

Permalink
Merge branch 'clustering_components_api' into sc2_recording_slices
Browse files Browse the repository at this point in the history
  • Loading branch information
yger committed Nov 13, 2024
2 parents 019b399 + 035d8d2 commit dfd9bff
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 5 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -84,9 +84,7 @@ def main_function(cls, recording, peaks, params, job_kwargs=dict()):
if params["peak_locations"] is None:
from spikeinterface.sortingcomponents.peak_localization import localize_peaks

peak_locations = localize_peaks(
recording, peaks, **params["peak_localization_kwargs"], **job_kwargs
)
peak_locations = localize_peaks(recording, peaks, **params["peak_localization_kwargs"], **job_kwargs)
else:
peak_locations = params["peak_locations"]

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -65,7 +65,7 @@ def main_function(cls, recording, peaks, params, job_kwargs=dict()):
peak_labels = cls._find_clusters(recording, peaks, wfs_arrays, sparsity_mask, noise, params)

wfs_arrays2, sparsity_mask2 = cls._prepare_clean(
recording, peaks, wfs_arrays, sparsity_mask, peak_labels, params
recording, peaks, wfs_arrays, sparsity_mask, peak_labels, params, job_kwargs
)

clean_peak_labels, peak_sample_shifts = cls._clean_cluster(
Expand Down Expand Up @@ -400,7 +400,7 @@ def _find_clusters(cls, recording, peaks, wfs_arrays, sparsity_mask, noise, d):
return peak_labels

@classmethod
def _prepare_clean(cls, recording, peaks, wfs_arrays, sparsity_mask, peak_labels, d):
def _prepare_clean(cls, recording, peaks, wfs_arrays, sparsity_mask, peak_labels, d, job_kwargs):
tmp_folder = d["tmp_folder"]
if tmp_folder is None:
wf_folder = None
Expand Down

0 comments on commit dfd9bff

Please sign in to comment.