From 4522522e1dc7f3f7a685bce103e9c6c2752ef233 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Fabian=20Fr=C3=B6hlich?= <fabian.frohlich@crick.ac.uk>
Date: Tue, 17 Dec 2024 19:59:23 +0000
Subject: [PATCH] remove eqx.debug.nan, fixes 2629 (#2630)

---
 python/sdist/amici/jax/petab.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/python/sdist/amici/jax/petab.py b/python/sdist/amici/jax/petab.py
index b5834223fb..b2b42f5c2a 100644
--- 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)),