fix: focusing on named_components

moverseai · Nov 14, 2024 · 6604134 · 6604134
1 parent b3fc252
commit 6604134
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/moai/engine/callbacks/model.py b/moai/engine/callbacks/model.py
@@ -16,12 +16,12 @@ def __init__(
     ):
         super().__init__(list)
         if model:
-            self.data.extend((c for c in model.children() if isinstance(c, Callback)))
-            for c in model.children():
-                for k, v in c.items():
-                    if isinstance(v, Callback):
-                        self.data.append(v)
+            # deprecated self.data.extend((c for c in model.children() if isinstance(c, Callback)))
 
+            if hasattr(model, "named_components"):
+                for component in model.named_components.values():
+                    if isinstance(component, Callback):
+                        self.data.append(component)
             if hasattr(model, "named_monitors"):
                 # search within named monitors
                 # for callbacks