From c1d32741c1b3b20006d9aefd8f212e4f9ce3c9f1 Mon Sep 17 00:00:00 2001
From: Andras Schmelczer <andras@schmelczer.dev>
Date: Fri, 1 Jul 2022 19:57:24 +0200
Subject: [PATCH] Improve CPU usage

---
 src/great_ai/utilities/parallel_map/mapper_function.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/great_ai/utilities/parallel_map/mapper_function.py b/src/great_ai/utilities/parallel_map/mapper_function.py
index 02526d2..c4650c1 100644
--- 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