feat: make add model element elegantly

smallbunnies · Jan 9, 2025 · 9fe0317 · 9fe0317
1 parent 29d928e
commit 9fe0317
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/Solverz/model/basic.py b/Solverz/model/basic.py
@@ -92,3 +92,13 @@ def create_instance(self):
             warnings.warn(f'Equation size {eqs.eqn_size} and variable size {eqs.vsize} not equal!')
 
         return eqs, y0
+
+    def add(self, m):
+        if isinstance(m, Model):
+            self.__dict__.update(m.__dict__)
+        elif isinstance(m, (Var, ParamBase, Eqn, Ode)):
+            self.__dict__[m.name] = m
+        elif isinstance(m, dict):
+            self.__dict__.update(m)
+        else:
+            raise ValueError(f"Unknown element type {type(m)}")