fix parameter unscaling

AMICI-dev · Dec 6, 2024 · 7b5383c · 7b5383c
1 parent 2be3ef6
commit 7b5383c
Showing 1 changed file with 11 additions and 2 deletions.
diff --git a/python/sdist/amici/jax/petab.py b/python/sdist/amici/jax/petab.py
@@ -401,9 +401,18 @@ def load_reinitialisation(
                     ],
                     Number,
                 )
-                else p
+                else p[self.model.parameter_ids.index(xval)]
                 if xval in self.model.parameter_ids
-                else self.get_petab_parameter_by_id(xval)
+                else jax_unscale(
+                    self.get_petab_parameter_by_id(xval),
+                    self._petab_problem.parameter_df.loc[
+                        xval, petab.PARAMETER_SCALE
+                    ],
+                )
+                if xval in self.parameter_ids
+                else self._petab_problem.parameter_df.loc[
+                    xval, petab.NOMINAL_VALUE
+                ]
                 for xname in self.model.state_ids
             ]
         )