From f1b3ee715b470b75c026f14c6c6e0877d5e29d23 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Fabian=20Fr=C3=B6hlich?= <fabian@schaluck.com>
Date: Wed, 18 Dec 2024 22:37:48 +0000
Subject: [PATCH] Update petab.py

---
 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 8db5055fdf..80acc9969a 100644
--- a/python/sdist/amici/jax/petab.py
+++ b/python/sdist/amici/jax/petab.py
@@ -547,7 +547,7 @@ def run_preequilibration(
             simulation_condition, p
         )
         return self.model.preequilibrate_condition(
-            p=eqx.debug.backward_nan(p),
+            p=p,
             mask_reinit=mask_reinit,
             x_reinit=x_reinit,
             solver=solver,