Update generate_model_latex.py (#4857)

DARPA-ASKEM · Sep 19, 2024 · 6d953ae · 6d953ae
1 parent 6125e5d
commit 6d953ae
Showing 1 changed file with 10 additions and 9 deletions.
diff --git a/packages/mira/tasks/generate_model_latex.py b/packages/mira/tasks/generate_model_latex.py
@@ -38,21 +38,22 @@ def main():
 
         t = sympy.Symbol(time)
 
-        # Observables
-        if len(model.observables) != 0:
-            obs_eqs = [
-                f"{{{obs.name}}}(t) = " + sympy.latex(obs.expression.args[0])
-                for obs in model.observables.values()
-            ]
-
         # Construct Sympy equations
         odesys = [
             sympy.latex(sympy.Eq(sympy.diff(sympy.Function(var)(t), t), terms))
             for var, terms in odeterms.items()
         ]
 
-        #add observables.
-        odesys += obs_eqs
+				# Observables
+				if len(model.observables) != 0:
+					obs_eqs = [
+						f"{{{obs.name}}}(t) = " + sympy.latex(obs.expression.args[0])
+						for obs in model.observables.values()
+					]
+
+					#add observables.
+					odesys += obs_eqs
+
 		    #Reformat:
         odesys = "\\begin{align} \n    " + " \\\\ \n    ".join([eq for eq in odesys]) + "\n\\end{align}"