diff --git a/src/quemb/kbe/pbe.py b/src/quemb/kbe/pbe.py index 583cbc64..9211f50f 100644 --- 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 index 7dcc9679..76856324 100644 --- 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