Also dill the creator methods.

OFFIS-DAI · Oct 18, 2024 · ee861a5 · ee861a5
1 parent 0146734
commit ee861a5
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/mango/container/mp.py b/mango/container/mp.py
@@ -103,6 +103,8 @@ def create_agent_process_environment(
     """
     asyncio.set_event_loop(asyncio.new_event_loop())
     container_data.codec = dill.loads(container_data.codec)
+    agent_creator = dill.loads(agent_creator)
+    mirror_container_creator = dill.loads(mirror_container_creator)
 
     async def start_agent_loop():
         container = mirror_container_creator(
@@ -453,8 +455,8 @@ def create_agent_process(self, agent_creator, container, mirror_container_creato
                         clock=container.clock,
                         kwargs=container._kwargs,
                     ),
-                    agent_creator,
-                    mirror_container_creator,
+                    dill.dumps(agent_creator),
+                    dill.dumps(mirror_container_creator),
                     to_pipe_message,
                     self._main_queue,
                     to_pipe,