diff --git a/dimos/core/core.py b/dimos/core/core.py index 6c95700926..1fac36d250 100644 --- a/dimos/core/core.py +++ b/dimos/core/core.py @@ -20,13 +20,9 @@ TypeVar, ) -from dimos.core.o3dpickle import register_picklers - if TYPE_CHECKING: from collections.abc import Callable -# injects pickling system into o3d -register_picklers() T = TypeVar("T") from typing import ParamSpec, TypeVar diff --git a/dimos/core/module_coordinator.py b/dimos/core/module_coordinator.py index 9ed3578731..a80c1b6f44 100644 --- a/dimos/core/module_coordinator.py +++ b/dimos/core/module_coordinator.py @@ -50,6 +50,9 @@ def __init__( self._deployed_modules = {} def start(self) -> None: + from dimos.core.o3dpickle import register_picklers + + register_picklers() for m in self._managers.values(): m.start()