removed collect_results for efficiency (fixed error)

Becksteinlab · Feb 9, 2025 · 5b82fae · 5b82fae
1 parent b9585c2
commit 5b82fae
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/basicrta/cluster.py b/basicrta/cluster.py
@@ -96,6 +96,16 @@ def get_taus(self):
         """
         from basicrta.util import get_bars
 
+        with (Pool(nproc, initializer=tqdm.set_lock,
+                   initargs=(Lock(),)) as p):
+            try:
+                for _ in tqdm(p.istarmap(self._single_residue, inarr),
+                              total=len(dirs), position=0,
+                              desc='overall progress'):
+                    pass
+            except KeyboardInterrupt:
+                pass
+
         taus = []
         for res in tqdm(self.residues, total=len(self.residues)):
             taus.append(res.tau)