Return None if unit is None

EnzymeML · Sep 24, 2024 · 4efc6d1 · 4efc6d1
1 parent de446b6
commit 4efc6d1
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/pyenzyme/sbml/serializer.py b/pyenzyme/sbml/serializer.py
@@ -438,9 +438,12 @@ def _get_sbml_kind(unit_type: pe.UnitType):
         raise ValueError(f"Unit type {unit_type} not found in libsbml")
 
 
-def _get_unit_id(unit: pe.UnitDefinition) -> str:
+def _get_unit_id(unit: pe.UnitDefinition) -> str | None:
     """Helper function to get the unit from the list of units."""
 
+    if unit is None:
+        return None
+
     if unit.id is None:
         raise ValueError(f"Unit {unit.name} does not have an ID")