modified main paths

Becksteinlab · Feb 22, 2024 · 81666ad · 81666ad
1 parent cc6427d
commit 81666ad
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/basicrta/gibbs.py b/basicrta/gibbs.py
@@ -99,7 +99,8 @@ def run(self, run_resids=None):
                                 self.ncomp, self.niter] for i in
                                range(len(residues))], dtype=object)
 
-        with Pool(self.nproc, initializer=tqdm.set_lock, initargs=(Lock(),)) as p:
+        with (Pool(self.nproc, initializer=tqdm.set_lock, initargs=(Lock(),)) as
+              p):
             try:
                 for _ in tqdm(p.istarmap(run_residue, input_list),
                               total=len(residues), position=0,