From 6814829c27cecf4619a5900368eafd3201f46ebf Mon Sep 17 00:00:00 2001 From: Rick Sexton Date: Fri, 1 Mar 2024 22:06:50 -0700 Subject: [PATCH] fixing local pickle error --- basicrta/gibbs.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/basicrta/gibbs.py b/basicrta/gibbs.py index 99e79db..f7b918a 100644 --- 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