FIX: use threading multiprocessing backend to still get proper stdouts

rochefort-lab · Jul 9, 2021 · acadd13 · acadd13
1 parent 8005d83
commit acadd13
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/fissa/core.py b/fissa/core.py
@@ -756,7 +756,7 @@ def separation_prep(self, redo=False):
 
         if use_multiprocessing:
             # run extraction
-            outputs = Parallel(n_jobs=self.ncores_separation)(
+            outputs = Parallel(n_jobs=self.ncores_separation, backend="threading")(
                 delayed(_extract_cfg)(self.images[i], self.rois[i])
                 for i in tqdm.tqdm(
                     range(self.nTrials),
@@ -922,7 +922,7 @@ def separate(self, redo_prep=False, redo_sep=False):
 
         # Do the extraction
         if use_multiprocessing:
-            outputs = Parallel(n_jobs=self.ncores_separation)(
+            outputs = Parallel(n_jobs=self.ncores_separation, backend="threading")(
                 delayed(_separate_cfg)(self.raw[i], i)
                 for i in tqdm.tqdm(
                     range(self.nCell),