Merge pull request #332 from ariddell/feature/issue-249

Avoid using context manager with multiprocessing.Pool
stan-dev · May 1, 2017 · 2316708 · 2316708
2 parents c6ad85d + 634d937
commit 2316708
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/pystan/model.py b/pystan/model.py
@@ -76,8 +76,12 @@ def _map_parallel(function, args, n_jobs):
     if multiprocessing and int(n_jobs) not in (0, 1):
         if n_jobs == -1:
             n_jobs = None
-        with multiprocessing.Pool(processes=n_jobs) as pool:
+        try:
+            pool = multiprocessing.Pool(processes=n_jobs)
             map_result = pool.map(function, args)
+        finally:
+            pool.close()
+            pool.join()
     else:
         map_result = list(map(function, args))
     return map_result