Wait for validation Pool to finish

artefactual-labs · Aug 30, 2024 · 902051d · 902051d
1 parent 5b77243
commit 902051d
Showing 1 changed file with 6 additions and 7 deletions.
diff --git a/bagit.py b/bagit.py
@@ -886,13 +886,12 @@ def _validate_entries(self, processes):
             if processes == 1:
                 hash_results = [_calc_hashes(i) for i in args]
             else:
-                try:
-                    pool = multiprocessing.Pool(
-                        processes if processes else None, initializer=worker_init
-                    )
-                    hash_results = pool.map(_calc_hashes, args)
-                finally:
-                    pool.terminate()
+                pool = multiprocessing.Pool(
+                    processes if processes else None, initializer=worker_init
+                )
+                hash_results = pool.map(_calc_hashes, args)
+                pool.close()
+                pool.join()
 
         # Any unhandled exceptions are probably fatal
         except: