From d414c2d6561966ca1979d11ce0a16c9029011fab Mon Sep 17 00:00:00 2001 From: Pierre Yger Date: Fri, 22 Nov 2024 09:56:24 +0100 Subject: [PATCH] Remove prints --- src/spikeinterface/sorters/internal/spyking_circus2.py | 2 -- src/spikeinterface/sortingcomponents/clustering/circus.py | 2 +- src/spikeinterface/sortingcomponents/clustering/main.py | 1 - 3 files changed, 1 insertion(+), 4 deletions(-) diff --git a/src/spikeinterface/sorters/internal/spyking_circus2.py b/src/spikeinterface/sorters/internal/spyking_circus2.py index f74570806c..208d9f5bc6 100644 --- a/src/spikeinterface/sorters/internal/spyking_circus2.py +++ b/src/spikeinterface/sorters/internal/spyking_circus2.py @@ -118,7 +118,6 @@ def _run_from_folder(cls, sorter_output_folder, params, verbose): job_kwargs = fix_job_kwargs(params["job_kwargs"]) job_kwargs.update({"progress_bar": verbose}) - print(job_kwargs) recording = cls.load_recording_from_folder(sorter_output_folder.parent, with_warnings=False) @@ -231,7 +230,6 @@ def _run_from_folder(cls, sorter_output_folder, params, verbose): else: clustering_method = "random_projections" - print('test') labels, peak_labels = find_cluster_from_peaks( recording_w, selected_peaks, method=clustering_method, method_kwargs=clustering_params, **job_kwargs ) diff --git a/src/spikeinterface/sortingcomponents/clustering/circus.py b/src/spikeinterface/sortingcomponents/clustering/circus.py index 32fe69ee38..6a341047f4 100644 --- a/src/spikeinterface/sortingcomponents/clustering/circus.py +++ b/src/spikeinterface/sortingcomponents/clustering/circus.py @@ -184,7 +184,7 @@ def main_function(cls, recording, peaks, params, job_kwargs=dict()): sparse_mask = node1.neighbours_mask neighbours_mask = get_channel_distances(recording) <= radius_um - # np.save(features_folder / "sparse_mask.npy", sparse_mask) + # np.save(features_folder / "sparse_mask.npy", sparse_mask) np.save(features_folder / "peaks.npy", peaks) original_labels = peaks["channel_index"] diff --git a/src/spikeinterface/sortingcomponents/clustering/main.py b/src/spikeinterface/sortingcomponents/clustering/main.py index fadcb07527..ba0fe6f9ac 100644 --- a/src/spikeinterface/sortingcomponents/clustering/main.py +++ b/src/spikeinterface/sortingcomponents/clustering/main.py @@ -32,7 +32,6 @@ def find_cluster_from_peaks(recording, peaks, method="stupid", method_kwargs={}, peak_labels.shape[0] == peaks.shape[0] """ job_kwargs = fix_job_kwargs(job_kwargs) - print("toto", job_kwargs) assert ( method in clustering_methods