wait

adrn · Sep 23, 2016 · f1e4ff6 · f1e4ff6
1 parent 9769f08
commit f1e4ff6
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/schwimmbad/__init__.py b/schwimmbad/__init__.py
@@ -33,6 +33,7 @@ def choose_pool(mpi=False, processes=1, **kwargs):
 
         pool = MPIPool(**kwargs)
         if not pool.is_master():
+            pool.wait()
             sys.exit(0)
 
         log.info("Running with MPI on {} cores".format(pool.size))