Add stopPool functionality

MatthewThe · Dec 15, 2023 · 8c15ad8 · 8c15ad8
1 parent 93fdfd3
commit 8c15ad8
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/job_pool/job_pool.py b/job_pool/job_pool.py
@@ -110,6 +110,10 @@ def checkPool(self, printProgressEvery: int = -1):
             self.pool.join()
             sys.exit(1)
 
+    def stopPool(self):
+        self.pool.terminate()
+        self.pool.join()
+
     def checkForTerminatedProcess(self, res, processes):
         start_time = time.time()
         while not res.ready():