Merge pull request #2 from moverseai/callback_fix

fix: update model.py to support callbacks for `named_components`
moverseai · Nov 19, 2024 · b56dfa3 · b56dfa3
2 parents ec3075b + 6604134
commit b56dfa3
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/moai/engine/callbacks/model.py b/moai/engine/callbacks/model.py
@@ -16,7 +16,12 @@ def __init__(
     ):
         super().__init__(list)
         if model:
-            self.data.extend((c for c in model.children() if isinstance(c, Callback)))
+            # 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