fixup

AMICI-dev · Feb 1, 2025 · 5170322 · 5170322
1 parent e1b5bc4
commit 5170322
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/python/sdist/amici/sbml_import.py b/python/sdist/amici/sbml_import.py
@@ -1990,7 +1990,9 @@ def _process_observables(
                 {
                     name
                     for sigma in sigmas.values()
-                    for symbol in sp.sympify(sigma).free_symbols
+                    for symbol in self._sympy_from_sbml_math(
+                        sigma
+                    ).free_symbols
                     if re.match(r"noiseParameter\d+$", (name := str(symbol)))
                 }
             )
@@ -2008,7 +2010,9 @@ def _process_observables(
                 {
                     name
                     for obs in observables.values()
-                    for symbol in sp.sympify(obs["formula"]).free_symbols
+                    for symbol in self._sympy_from_sbml_math(
+                        obs["formula"]
+                    ).free_symbols
                     if re.match(
                         r"observableParameter\d+$", (name := str(symbol))
                     )

diff --git a/python/tests/test_jax.py b/python/tests/test_jax.py
@@ -194,6 +194,8 @@ def check_fields_jax(
         "ts_posteq": jnp.array(ts_posteq),
         "my": jnp.array(my),
         "iys": jnp.array(iys),
+        "ops": jnp.zeros((*my.shape[:2], 0)),
+        "nps": jnp.zeros((*my.shape[:2], 0)),
         "iy_trafos": jnp.array(iy_trafos),
         "x_preeq": jnp.array([]),
         "solver": diffrax.Kvaerno5(),