From cb77f756121daf0bd896975265da6ac2c37073bf Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Beno=C3=AEt=20Sierro?= Date: Mon, 21 Jun 2021 17:17:49 +0200 Subject: [PATCH] better merge pbar --- src/scgenerator/io.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/scgenerator/io.py b/src/scgenerator/io.py index 2b372dd..1018f16 100644 --- a/src/scgenerator/io.py +++ b/src/scgenerator/io.py @@ -411,12 +411,12 @@ def merge(destination: os.PathLike, path_trees: List[PathTree] = None): z_num += prev_z_num pbars = utils.PBars( - path_trees, 1, desc="Merging", worker_kwargs=dict(total=z_num, desc="current pos") + len(path_trees) * z_num, "Merging", 1, worker_kwargs=dict(total=z_num, desc="current pos") ) for path_tree in path_trees: + pbars.reset(1) iden = PARAM_SEPARATOR.join(path_tree[-1][0].name.split()[2:-2]) merge_path_tree(path_tree, destination / iden, z_callback=lambda i: pbars.update(1)) - pbars.update(0) def sim_dirs(path_trees: List[PathTree]) -> Generator[Path, None, None]: