Fix handling of mathml serialization

gyorilab · Jun 23, 2023 · 50ab4a5 · 50ab4a5
1 parent 041d696
commit 50ab4a5
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/mira/modeling/askenet/petrinet.py b/mira/modeling/askenet/petrinet.py
@@ -75,7 +75,7 @@ def __init__(self, model: Model):
                 states_dict['units'] = {
                     'expression': str(var.concept.units.expression),
                     'expression_mathml': expression_to_mathml(
-                        var.concept.units.expression),
+                        var.concept.units.expression.args[0]),
                 }
 
             self.states.append(states_dict)
@@ -102,7 +102,7 @@ def __init__(self, model: Model):
                 'name': observable.observable.name,
                 'expression': str(observable.observable.expression),
                 'expression_mathml': expression_to_mathml(
-                    observable.observable.expression),
+                    observable.observable.expression.args[0]),
             }
             self.observables.append(obs_data)
 
@@ -112,7 +112,7 @@ def __init__(self, model: Model):
                 self.time['units'] = {
                     'expression': str(model.template_model.time.units.expression),
                     'expression_mathml': expression_to_mathml(
-                        model.template_model.time.units.expression),
+                        model.template_model.time.units.expression.args[0]),
                 }
         else:
             self.time = None
@@ -186,7 +186,7 @@ def __init__(self, model: Model):
                 param_dict['units'] = {
                     'expression': str(param.concept.units.expression),
                     'expression_mathml': expression_to_mathml(
-                        param.concept.units.expression),
+                        param.concept.units.expression.args[0]),
                 }
             self.parameters.append(param_dict)