SBML import - faster _get_list_of_species_references (#2094)

AMICI-dev · May 17, 2023 · d0a12c4 · d0a12c4
1 parent 52a1331
commit d0a12c4
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/python/sdist/amici/sbml_import.py b/python/sdist/amici/sbml_import.py
@@ -2535,11 +2535,11 @@ def _get_list_of_species_references(
         ListOfSpeciesReferences
     """
     return [
-        reference
-        for element in sbml_model.all_elements
-        if isinstance(element, sbml.ListOfSpeciesReferences)
-        for reference in element
-    ]
+            reference
+            for reaction in sbml_model.getListOfReactions()
+            for reference in
+            itt.chain(reaction.getListOfReactants(), reaction.getListOfProducts(), reaction.getListOfModifiers())
+        ]
 
 
 def replace_logx(math_str: Union[str, float, None]) -> Union[str, float, None]: