Close pool

bilby/core/sampler/ptemcee.py

@@ -124,7 +124,9 @@ class Ptemcee(MCMCSampler):
             with schwimmbad.MultiPool(self.threads, initializer=init,
                                       initargs=(self.likelihood, self.priors)) as pool:
                 self.pool = pool
-                return self.run_sampler_internal()
+                out = self.run_sampler_internal()
+                self.pool.close()
+                return out
         else:
             self.pool = None
             return self.run_sampler_internal()
@@ -257,6 +259,8 @@ class Ptemcee(MCMCSampler):
 
     def write_current_state_and_exit(self, signum=None, frame=None):
         logger.warning("Run terminated with signal {}".format(signum))
+        if self.pool:
+            self.pool.close()
         self.write_current_state()
         sys.exit(77)