Merge branch 'diffrax_update' of https://github.com/AMICI-dev/AMICI i…

…nto diffrax_update
AMICI-dev · Dec 18, 2024 · 990d385 · 990d385
2 parents 7087f5f + d29dc5c
commit 990d385
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/sdist/amici/jax/petab.py b/python/sdist/amici/jax/petab.py
@@ -495,7 +495,7 @@ def run_simulation(
             simulation_condition[0], p
         )
         return self.model.simulate_condition(
-            p=eqx.debug.backward_nan(p),
+            p=p,
             ts_dyn=jax.lax.stop_gradient(jnp.array(ts_dyn)),
             ts_posteq=jax.lax.stop_gradient(jnp.array(ts_posteq)),
             my=jax.lax.stop_gradient(jnp.array(my)),