fixing local pickle error

Becksteinlab · Mar 2, 2024 · 6814829 · 6814829
1 parent 864c1f7
commit 6814829
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/basicrta/gibbs.py b/basicrta/gibbs.py
@@ -52,11 +52,10 @@ def collect_results(self, nproc=1):
         dirs = np.array(glob('?[0-9]*'))
         sorted_inds = np.array([int(adir[1:]) for adir in dirs]).argsort()
         dirs = dirs[sorted_inds]
-        print(dirs)
-        with (Pool(nproc, initializer=tqdm.set_lock,
-                   initargs=(Lock(),)) as p):
+        print(len(dirs))
+        with Pool(nproc, initializer=tqdm.set_lock, initargs=(Lock(),)) as p:
             try:
-                for _ in tqdm(p.starmap(self._collect_res, dirs),
+                for _ in tqdm(p.map(self._collect_res, dirs),
                               total=len(dirs), position=0,
                               desc='overall progress'):
                     pass