diff --git a/dimos/core/global_config.py b/dimos/core/global_config.py index 69bd33beb8..9b68a80384 100644 --- a/dimos/core/global_config.py +++ b/dimos/core/global_config.py @@ -22,6 +22,7 @@ class GlobalConfig(BaseSettings): simulation: bool = False replay: bool = False n_dask_workers: int = 2 + memory_limit: str = "auto" mujoco_room: str | None = None robot_model: str | None = None diff --git a/dimos/core/module_coordinator.py b/dimos/core/module_coordinator.py index 10eb425d35..d29e7968c6 100644 --- a/dimos/core/module_coordinator.py +++ b/dimos/core/module_coordinator.py @@ -32,12 +32,11 @@ class ModuleCoordinator(Resource): def __init__( self, n: int | None = None, - memory_limit: str = "auto", global_config: GlobalConfig | None = None, ) -> None: cfg = global_config or GlobalConfig() self._n = n if n is not None else cfg.n_dask_workers - self._memory_limit = memory_limit + self._memory_limit = cfg.memory_limit def start(self) -> None: self._client = core.start(self._n, self._memory_limit)