BUG: use ThreadPoolExecutor instead of ThreadPool

Use `concurrent.futures.ThreadPoolExecutor` in distutils instead of `multiprocessing.pool.ThreadPool`. Fix numpy#21026
melissawm · Apr 12, 2022 · f2a965c · f2a965c
1 parent 55affcb
commit f2a965c
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/numpy/distutils/ccompiler.py b/numpy/distutils/ccompiler.py
@@ -355,10 +355,9 @@ def single_compile(args):
 
     if len(build) > 1 and jobs > 1:
         # build parallel
-        import multiprocessing.pool
-        pool = multiprocessing.pool.ThreadPool(jobs)
-        pool.map(single_compile, build_items)
-        pool.close()
+        from concurrent.futures import ThreadPoolExecutor
+        with ThreadPoolExecutor(jobs) as pool:
+            pool.map(single_compile, build_items)
     else:
         # build serial
         for o in build_items: