[core] switch to multiprocessing.Context for mp calls. fixes #51

Author: terhorst

smcpp/commands/posterior.py

@@ -8,7 +8,6 @@
 import scipy.optimize
 import scipy.ndimage
 import pprint
-import multiprocessing
 import sys
 from collections import Counter
 import sys

smcpp/data_filter.py

@@ -4,14 +4,15 @@
 import sys
 import wrapt
 from concurrent.futures import ThreadPoolExecutor
-import multiprocessing
-import multiprocessing.dummy
 from collections import OrderedDict
 import contextlib
+import multiprocessing
+
 
 from . import logging, estimation_tools, defaults
 
 logger = logging.getLogger(__name__)
+mp_ctx = multiprocessing.get_context('forkserver')
 
 @attr.s
 class DataPipeline:
@@ -68,7 +69,7 @@ def __call__(self, contigs):
 
 @attr.s
 class ProcessParallelFilter(ParallelFilter):
-    Pool = multiprocessing.Pool
+    Pool = mp_ctx.Pool
 
 @attr.s
 class ThreadParallelFilter(ParallelFilter):

smcpp/frontend/console.py

@@ -1,4 +1,3 @@
-import multiprocessing as mp
 from argparse import ArgumentParser
 
 from .. import commands, logging, version, defaults, _smcpp
@@ -19,7 +18,6 @@ def main():
     logging.init_logging()
     logger = logging.getLogger(__name__)
     logger.debug("SMC++ " + version.version)
-    mp.set_start_method('forkserver')
     parser = ArgumentParser()
     subparsers = parser.add_subparsers(dest='command')
     subparsers.required = True