[Typing][Fix] Ensure all workers are running complete before pool ter…

…minate (#65199)
PaddlePaddle · Jun 15, 2024 · 8261cde · 8261cde
1 parent 2e66576
commit 8261cde
Showing 1 changed file with 9 additions and 7 deletions.
diff --git a/tools/type_checking.py b/tools/type_checking.py
@@ -262,13 +262,15 @@ def get_test_results(
             )
 
     test_results = []
-    pool = multiprocessing.Pool(initializer=init_worker)
-    try:
-        test_results = pool.starmap(type_checker.run, codeblocks)
-
-    except KeyboardInterrupt:
-        pool.terminate()
-        pool.join()
+    with multiprocessing.Pool(initializer=init_worker) as pool:
+        try:
+            test_results = pool.starmap(type_checker.run, codeblocks)
+        except KeyboardInterrupt:
+            pool.terminate()
+        else:
+            pool.close()
+        finally:
+            pool.join()
 
     return list(test_results)