diff --git a/mira/modeling/askenet/petrinet.py b/mira/modeling/askenet/petrinet.py index d764e371a..469413cc1 100644 --- 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)