From 50db7b8cb25891cc6ec09d471e616f03d18cb702 Mon Sep 17 00:00:00 2001 From: Andras Schmelczer Date: Mon, 4 Jul 2022 23:09:03 +0200 Subject: [PATCH] Fix batch tests --- great_ai/utilities/parallel_map/parallel_map.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/great_ai/utilities/parallel_map/parallel_map.py b/great_ai/utilities/parallel_map/parallel_map.py index 0c088c0..e95604d 100644 --- a/great_ai/utilities/parallel_map/parallel_map.py +++ b/great_ai/utilities/parallel_map/parallel_map.py @@ -99,7 +99,7 @@ def parallel_map( output_queue = manager.Queue(config.concurrency * 2) should_stop = ctx.Event() - serialized_map_function = dill.dumps(function, byref=True, recurse=True) + serialized_map_function = dill.dumps(function, byref=True, recurse=False) processes = [ ctx.Process(