join pool after constructing covariance matrix

mir-group · Jun 16, 2019 · 5dd377a · 5dd377a
1 parent c26a190
commit 5dd377a
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/flare/gp_algebra.py b/flare/gp_algebra.py
@@ -76,6 +76,7 @@ def get_ky_mat_par(hyps: np.ndarray, training_data: list,
     ky_mat = k_mat + sigma_n ** 2 * np.eye(size)
 
     pool.close()
+    pool.join()
 
     return ky_mat
 
@@ -124,6 +125,7 @@ def get_ky_and_hyp_par(hyps: np.ndarray, training_data: list,
     ky_mat = k_mat + sigma_n ** 2 * np.eye(size)
 
     pool.close()
+    pool.join()
 
     return hyp_mat, ky_mat