hot bug fixed

dubovikmaster · Jul 24, 2022 · 015cf2e · 015cf2e
1 parent 8acb1b1
commit 015cf2e
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/parallelbar/parallelbar.py b/parallelbar/parallelbar.py
@@ -178,7 +178,7 @@ def progress_imap(func, tasks, initializer=None, initargs=(), n_cpu=None, chunk_
     if isinstance(tasks, abc.Iterator) and not total:
         raise ValueError('If the tasks are an iterator, the total parameter must be specified')
     if process_timeout:
-        func = partial(_wrapped_func, func, process_timeout)
+        func = partial(_wrapped_func, func, process_timeout, True)
     result = _do_parallel(func, 'imap', tasks, initializer, initargs, n_cpu, chunk_size, core_progress, context, total,
                           bar_step, disable,
                           None)
@@ -193,7 +193,7 @@ def progress_imapu(func, tasks, initializer=None, initargs=(), n_cpu=None, chunk
     if isinstance(tasks, abc.Iterator) and not total:
         raise ValueError('If the tasks are an iterator, the total parameter must be specified')
     if process_timeout:
-        func = partial(_wrapped_func, func, process_timeout)
+        func = partial(_wrapped_func, func, process_timeout, True)
     result = _do_parallel(func, 'imap_unordered', tasks, initializer, initargs, n_cpu, chunk_size, core_progress,
                           context, total, bar_step,
                           disable, None)

diff --git a/setup.py b/setup.py
@@ -6,7 +6,7 @@
 
 setup(
     name='parallelbar',
-    version='0.2.11',
+    version='0.2.12',
     packages=find_packages(),
     author='Dubovik Pavel',
     author_email='geometryk@gmail.com',