Defer o3dpickle

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

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()