changes for ms5 params

SpikeInterface · Apr 10, 2024 · fcbb607 · fcbb607
1 parent 1173f16
commit fcbb607
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 3 deletions.
diff --git a/src/spikeinterface_pipelines/pipeline.py b/src/spikeinterface_pipelines/pipeline.py
@@ -72,8 +72,6 @@ def run_pipeline(
     if isinstance(spikesorting_params, dict):
         if spikesorting_params["sorter_name"] not in sorter_model_map:
             raise ValueError(f"Sorter name {spikesorting_params['sorter_name']} not recognized")
-        if spikesorting_params["sorter_name"] == "mountainsort5":
-            spikesorting_params["sorter_kwargs"]["n_jobs_for_preprocessing"] = job_kwargs.n_jobs
         spikesorting_params = SpikeSortingParams(
             sorter_name=spikesorting_params["sorter_name"],
             sorter_kwargs=sorter_model_map[spikesorting_params["sorter_name"]](**spikesorting_params["sorter_kwargs"]),

diff --git a/src/spikeinterface_pipelines/spikesorting/params.py b/src/spikeinterface_pipelines/spikesorting/params.py
@@ -84,7 +84,6 @@ class MountainSort5Model(BaseModel):
     freq_max: int = Field(default=6000, description="Low-pass filter cutoff frequency")
     filter: bool = Field(default=True, description="Enable or disable filter")
     whiten: bool = Field(default=True, description="Enable or disable whiten")
-    n_jobs_for_preprocessing: float = Field(default=0.8, description="Number of jobs for preprocessing")
 
 
 ## SpykingCircus2 - WIP