Merge branch 'develop' into diffrax_update

AMICI-dev · Dec 18, 2024 · d29dc5c · d29dc5c
2 parents aba0d3d + 4522522
commit d29dc5c
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)),