diff --git a/common/models.py b/common/models.py index 9e22482..a3136e8 100644 --- a/common/models.py +++ b/common/models.py @@ -27,6 +27,11 @@ class JobKwargs(BaseModel): n_jobs: float = Field(default=0.8, description='Number of jobs, must be a positive number between 0 and 1, or -1 for all processors.') chunk_duration: str = Field(default='1s', description='Chunk duration.') progress_bar: bool = Field(default=False, description='Show progress bar.') + mp_context: str = Field( + default="spawn", + description='Context for multiprocessing. It can be "fork" or "spawn".', + json_schema_extra={'options': ["fork", "spawn"]}, + ) # ------------------------------ diff --git a/si_mountainsort5/requirements.txt b/si_mountainsort5/requirements.txt index 7871224..580a8f4 100644 --- a/si_mountainsort5/requirements.txt +++ b/si_mountainsort5/requirements.txt @@ -2,6 +2,6 @@ pynwb pyvips==2.2.2 neuroconv mountainsort5==0.5.6 -# spikeinterface[full]==0.100.5 -# spikeinterface_pipelines -# dendro==0.2.14 +# spikeinterface[full]>=0.100.5 +# spikeinterface_pipelines>=0.0.11 +# dendro>=0.2.14