From 757ffa1af197780b85eff50dd646bda3becb0002 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Fabian=20Fr=C3=B6hlich?= Date: Thu, 5 Dec 2024 23:22:35 +0000 Subject: [PATCH] fix state ids --- python/sdist/amici/jax/ode_export.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/sdist/amici/jax/ode_export.py b/python/sdist/amici/jax/ode_export.py index cec5104ded..4329195441 100644 --- a/python/sdist/amici/jax/ode_export.py +++ b/python/sdist/amici/jax/ode_export.py @@ -226,7 +226,7 @@ def _generate_jax_code(self) -> None: # assign named variables from a jax array **_jax_variable_assignments(self.model, sym_names), # tuple of variable names (ids as they are unique) - **_jax_variable_ids(self.model, ("p", "k", "y", "x")), + **_jax_variable_ids(self.model, ("p", "k", "y", "x_rdata")), **{ "MODEL_NAME": self.model_name, # keep track of the API version that the model was generated with so we