Merge branch 'main' of https://github.com/moverseai/moai

moai/engine/callbacks/model.py

@@ -16,12 +16,16 @@ def __init__(
     ):
         super().__init__(list)
         if model:
-            self.data.extend((c for c in model.children() if isinstance(c, Callback)))
-            # check for monads in the model
-            for flow in model.named_flows.values():
-                for module in flow.values():
-                    if isinstance(module, Callback):
-                        self.data.append(module)
+            if hasattr(model, "named_flows"):  # check for monads in the model
+                for flow in model.named_flows.values():
+                    for module in flow.values():
+                        if isinstance(module, Callback):
+                            self.data.append(module)
+
+            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