Improve CPU usage

2022-07-01 19:57:24 +02:00 · 2022-07-01 19:57:24 +02:00 · 06cd7de379
commit 06cd7de379
parent dac4ce7b7c
1 changed files with 2 additions and 2 deletions
--- a/src/great_ai/utilities/parallel_map/mapper_function.py
+++ b/src/great_ai/utilities/parallel_map/mapper_function.py
@ -16,10 +16,10 @@ def mapper_function(
    try:
        while not should_stop.is_set():
            try:
-                input_chunk = input_queue.get_nowait()
+                input_chunk = input_queue.get(1)
                output_chunk = [(i, map_function(v)) for i, v in input_chunk]
                output_queue.put(output_chunk)
            except queue.Empty:
                pass
-    except KeyboardInterrupt:
+    except (KeyboardInterrupt, BrokenPipeError):
        return