remove global multiprocessing.set_start_method (#8343)

spacetelescope · Apr 11, 2024 · fb8605c · fb8605c
1 parent 8ac7060
commit fb8605c
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 7 deletions.
diff --git a/jwst/jump/jump_step.py b/jwst/jump/jump_step.py
@@ -4,12 +4,9 @@
 from ..stpipe import Step
 from .jump import run_detect_jumps
 import time
-import multiprocessing
 
 __all__ = ["JumpStep"]
 
-multiprocessing.set_start_method('forkserver', force=True)
-
 
 class JumpStep(Step):
     """

diff --git a/jwst/ramp_fitting/ramp_fit_step.py b/jwst/ramp_fitting/ramp_fit_step.py
@@ -18,12 +18,10 @@
 import logging
 import copy
 import warnings
-import multiprocessing
 
 log = logging.getLogger(__name__)
 log.setLevel(logging.DEBUG)
 
-multiprocessing.set_start_method('forkserver', force=True)
 
 
 __all__ = ["RampFitStep"]

diff --git a/jwst/wfss_contam/observations.py b/jwst/wfss_contam/observations.py
@@ -1,6 +1,6 @@
 import time
+import multiprocessing
 import numpy as np
-from multiprocessing import Pool
 
 from scipy import sparse
 
@@ -259,7 +259,8 @@ def disperse_chunk(self, c, order, wmin, wmax, sens_waves, sens_resp):
 
         time1 = time.time()
         if self.max_cpu > 1:
-            mypool = Pool(self.max_cpu)  # Create the pool
+            ctx = multiprocessing.get_context("forkserver")
+            mypool = ctx.Pool(self.max_cpu)  # Create the pool
             all_res = mypool.imap_unordered(dispersed_pixel, pars)  # Fill the pool
             mypool.close()  # Drain the pool
         else: