mp context

catalystneuro · Apr 25, 2024 · 23f2921 · 23f2921
1 parent e4446b2
commit 23f2921
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git 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
@@ -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