Fix check on Python 2 systems

salilab · Jun 4, 2024 · 06415df · 06415df
1 parent 839ffc9
commit 06415df
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/modules/multifit/pyext/src/fit_fft.py b/modules/multifit/pyext/src/fit_fft.py
@@ -26,7 +26,8 @@
 def _get_context():
     # Use 'forkserver' rather than 'fork' start method if we can; 'fork' does
     # not work well with multithreaded processes or CUDA
-    if 'forkserver' in multiprocessing.get_all_start_methods():
+    if (hasattr(multiprocessing, 'get_all_start_methods')
+            and 'forkserver' in multiprocessing.get_all_start_methods()):
         return multiprocessing.get_context('forkserver')
     else:
         return multiprocessing.get_context()