Merge pull request #140 from OFFIS-DAI/allow_setting_mp_method

Add option to set multiprocessing method through mp_method param in container

mango/container/core.py

@@ -32,6 +32,7 @@ def __init__(
         clock: Clock,
         copy_internal_messages=False,
         mirror_data=None,
+        mp_method="spawn",
         **kwargs,
     ):
         self.name: str = name
@@ -64,7 +65,9 @@ def __init__(
                 self, self._mirror_data
             )
         else:
-            self._container_process_manager = MainContainerProcessManager(self)
+            self._container_process_manager = MainContainerProcessManager(
+                self, mp_method
+            )
 
     def _all_aids(self):
         all_aids = list(self._agents.keys()) + self._container_process_manager.aids

mango/container/mp.py

@@ -1,6 +1,7 @@
 import asyncio
 import logging
 import os
+import warnings
 from collections.abc import Callable
 from dataclasses import dataclass
 from multiprocessing import get_context
@@ -362,11 +363,12 @@ class MainContainerProcessManager(BaseContainerProcessManager):
     def __init__(
         self,
         container,
+        mp_method: str = "spawn",
     ) -> None:
         self._active = False
         self._container = container
         self._mp_enabled = False
-        self._ctx = get_context("spawn")
+        self._ctx = get_context(mp_method)
         self._agent_process_init_list = []
         self._started = False
 
@@ -510,7 +512,16 @@ def _create_agent_process_bytes(
         from_pipe_message, to_pipe_message = aioduplex(self._ctx)
         from_pipe, to_pipe = aioduplex(self._ctx)
         process_initialized = self._ctx.Event()
-        with to_pipe.detach() as to_pipe, to_pipe_message.detach() as to_pipe_message:
+        with (
+            warnings.catch_warnings(),
+            to_pipe.detach() as to_pipe,
+            to_pipe_message.detach() as to_pipe_message,
+        ):
+            warnings.filterwarnings(
+                "ignore",
+                message=r".*This process .* is multi-threaded, use of fork\(\) may lead to deadlocks.*",
+                category=DeprecationWarning,
+            )
             agent_process = self._ctx.Process(
                 target=create_agent_process_environment,
                 args=(