Add threaded parallel_map

2022-06-30 17:22:16 +02:00 · 2022-06-30 17:22:16 +02:00 · 0a91cc7287
commit 0a91cc7287
parent 30d148078c
1 changed files with 225 additions and 6 deletions
--- a/src/great_ai/utilities/parallel_map/parallel_map.py
+++ b/src/great_ai/utilities/parallel_map/parallel_map.py
@ -128,7 +128,7 @@ def parallel_map(
                pass

        should_stop.set()
-    except KeyboardInterrupt:
+    except Exception:
        for p in processes:
            p.terminate()
    finally: