Improve serialization

2022-07-03 12:25:56 +02:00 · 2022-07-03 12:25:56 +02:00 · c76342534a
commit c76342534a
parent 7d774f37be
1 changed files with 16 additions and 14 deletions
--- a/src/great_ai/utilities/parallel_map/threaded_parallel_map.py
+++ b/src/great_ai/utilities/parallel_map/threaded_parallel_map.py
@ -99,7 +99,7 @@ def threaded_parallel_map(
                input_queue=input_queue,
                output_queue=output_queue,
                should_stop=should_stop,
-                serialized_map_function=config.serialized_map_function,
+                map_function=function,
            ),
        )
        for i in range(config.concurrency)