Add description attribute to State class for petrinet amr, remove tra…

…iling comma
gyorilab · Oct 23, 2024 · 997d55c · 997d55c
1 parent f54dc4e
commit 997d55c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/mira/modeling/amr/petrinet.py b/mira/modeling/amr/petrinet.py
@@ -79,8 +79,7 @@ def __init__(self, model: Model):
                                     var.concept.identifiers.items()
                                     if k != 'biomodels.species'},
                     'modifiers': var.concept.context,
-                },
-
+                }
             if var.concept.units:
                 states_dict['units'] = {
                     'expression': str(var.concept.units.expression),
@@ -408,6 +407,7 @@ class Units(BaseModel):
 class State(BaseModel):
     id: str
     name: Optional[str] = None
+    description: Optional[str] = None
     grounding: Optional[Dict] = None
     units: Optional[Units] = None