From 7be0ed82bc5d1d7f38585da1cb0369b074239deb Mon Sep 17 00:00:00 2001
From: Sam Bull <Sam.B@snowfalltravel.com>
Date: Tue, 31 Mar 2026 13:43:26 +0100
Subject: [PATCH] Defer o3dpickle

---
 dimos/core/core.py               | 4 ----
 dimos/core/module_coordinator.py | 3 +++
 2 files changed, 3 insertions(+), 4 deletions(-)

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