From 58cdfaaf144de2f23f3abebb3a8875fe2642d023 Mon Sep 17 00:00:00 2001 From: Thinh Nguyen Date: Fri, 9 May 2025 07:02:27 -0500 Subject: [PATCH] chore: minor code optimization --- element_array_ephys/spike_sorting/si_spike_sorting.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/element_array_ephys/spike_sorting/si_spike_sorting.py b/element_array_ephys/spike_sorting/si_spike_sorting.py index e3e797b6..a324afa3 100644 --- a/element_array_ephys/spike_sorting/si_spike_sorting.py +++ b/element_array_ephys/spike_sorting/si_spike_sorting.py @@ -275,7 +275,7 @@ def make(self, key): postprocessing_params = params["SI_POSTPROCESSING_PARAMS"] - job_kwargs = postprocessing_params.get( + job_kwargs = postprocessing_params.pop( "job_kwargs", {"n_jobs": -1, "chunk_duration": "1s"} ) @@ -366,7 +366,7 @@ def make(self, key): postprocessing_params = params["SI_POSTPROCESSING_PARAMS"] - job_kwargs = postprocessing_params.get( + job_kwargs = postprocessing_params.pop( "job_kwargs", {"n_jobs": -1, "chunk_duration": "1s"} ) @@ -383,6 +383,8 @@ def _export_to_phy(): sorting_analyzer=sorting_analyzer, output_folder=analyzer_output_dir / "phy", use_relative_path=True, + remove_if_exists=True, + copy_binary=True, **job_kwargs, ) @@ -395,6 +397,7 @@ def _export_report(): si.exporters.export_report( sorting_analyzer=sorting_analyzer, output_folder=analyzer_output_dir / "spikeinterface_report", + remove_if_exists=True, **job_kwargs, )