Handle dimensionless

gyorilab · Jun 28, 2023 · cdfbd81 · cdfbd81
1 parent 89695dd
commit cdfbd81
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/mira/sources/sbml/processor.py b/mira/sources/sbml/processor.py
@@ -172,10 +172,6 @@ def _lookup_concepts_filtered(species_ids) -> List[Concept]:
             for species in self.sbml_model.species
         }
 
-        species_units = {
-            species.id: self.units[species.units] for species in self.sbml_model.species
-        }
-
         all_locals = {k: v for k, v in (list(parameter_symbols.items()) +
                                         list(compartment_symbols.items()) +
                                         list(function_lambdas.items()) +
@@ -341,7 +337,10 @@ def _extract_concepts(self) -> Mapping[str, Concept]:
 
     def get_object_units(self, object):
         if object.units:
-            return Unit(expression=self.units[object.units])
+            if object.units == 'dimensionless':
+                return Unit(expression=sympy.Integer(1))
+            else:
+                return Unit(expression=self.units[object.units])
         else:
             return None