fixup

Author: FFroehlich

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))
                     )

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(),