From fb57ec06ae5536d8db66b9f6af6d7cde37995074 Mon Sep 17 00:00:00 2001 From: "Azalea (on HyDEV-Daisy)" Date: Thu, 25 Aug 2022 23:12:50 -0400 Subject: [PATCH] [F] Fix bugs in tqdm utils --- hypy_utils/tqdm_utils.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/hypy_utils/tqdm_utils.py b/hypy_utils/tqdm_utils.py index 1095fcc..76cdf6c 100644 --- a/hypy_utils/tqdm_utils.py +++ b/hypy_utils/tqdm_utils.py @@ -12,17 +12,19 @@ from tqdm.contrib.concurrent import process_map, thread_map def smap(fn: Callable, lst: Iterable, *args, **kwargs) -> list: - return [fn(i) for i in tqdm.tqdm(lst, position=0, leave=True)] + return [fn(i) for i in tqdm.tqdm(lst, position=0, leave=True, *args, **kwargs)] def pmap(fn: Callable, lst: Iterable, *args, **kwargs) -> list: tqdm_args = dict(position=0, leave=True, chunksize=1, tqdm_class=tqdm.tqdm, max_workers=os.cpu_count()) - return process_map(fn, lst, *args, **{**tqdm_args, **kwargs}) + tqdm_args.update(kwargs) + return process_map(fn, lst, *args, **tqdm_args) def tmap(fn: Callable, lst: Iterable, *args, **kwargs) -> list: tqdm_args = dict(position=0, leave=True, chunksize=1, tqdm_class=tqdm.tqdm, max_workers=os.cpu_count()) - return thread_map(fn, lst, *args, **{**tqdm_args, **kwargs}) + tqdm_args.update(kwargs) + return thread_map(fn, lst, *args, **tqdm_args) def tq(it: Iterable, desc: str, *args, **kwargs) -> tqdm: