remove eqx.debug.nan, fixes 2629 (#2630)

AMICI-dev · Dec 17, 2024 · 45f64cb · 45f64cb
1 parent 3206fce
commit 45f64cb
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
@@ -500,7 +500,7 @@ def run_simulation(
             simulation_condition[0], p
         )
         return self.model.simulate_condition(
-            p=eqx.debug.backward_nan(p),
+            p=p,
             ts_init=jax.lax.stop_gradient(jnp.array(ts_preeq)),
             ts_dyn=jax.lax.stop_gradient(jnp.array(ts_dyn)),
             ts_posteq=jax.lax.stop_gradient(jnp.array(ts_posteq)),