use Threadpool

troyvvgroup · Jan 13, 2025 · 87b62d3 · 87b62d3
1 parent 0eefb8c
commit 87b62d3
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/quemb/kbe/pbe.py b/src/quemb/kbe/pbe.py
@@ -2,7 +2,7 @@
 
 import os
 import pickle
-from multiprocessing import Pool
+from multiprocessing.pool import ThreadPool as Pool
 
 import h5py
 import numpy as np

diff --git a/src/quemb/molbe/be_parallel.py b/src/quemb/molbe/be_parallel.py
@@ -1,7 +1,7 @@
 # Author(s): Oinam Romesh Meitei, Leah Weisburn
 
 import os
-from multiprocessing import Pool
+from multiprocessing.pool import ThreadPool as Pool
 
 from numpy import asarray, diag_indices, einsum, float64, zeros_like
 from numpy.linalg import multi_dot