Refactor

2022-07-13 08:33:22 +02:00 · 2022-07-13 08:33:22 +02:00 · 6106816999
commit 6106816999
parent 0585a4e4f9
1 changed files with 8 additions and 7 deletions
--- a/great_ai/utilities/parallel_map/simple_parallel_map.py
+++ b/great_ai/utilities/parallel_map/simple_parallel_map.py
@ -46,15 +46,16 @@ def simple_parallel_map(
    """

    input_values = list(input_values)  # in case the input is mistakenly not a sequence
+    generator = parallel_map(
+        func=func,
+        input_values=input_values,
+        chunk_size=chunk_size,
+        concurrency=concurrency,
+    )
+
    return list(
        tqdm(
-            parallel_map(
-                func=func,
-                input_values=input_values,
-                chunk_size=chunk_size,
-                concurrency=concurrency,
-            ),
+            generator,
            total=len(input_values),
-            dynamic_ncols=True,
        )
    )