From 3f05c7e51eed44eb7b693f331dc232817e1c0dc8 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Fabian=20Fr=C3=B6hlich?= <fabian@schaluck.com>
Date: Mon, 2 Dec 2024 22:47:07 +0000
Subject: [PATCH] fix parameter ids

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

diff --git a/python/sdist/amici/jax/jax.template.py b/python/sdist/amici/jax/jax.template.py
index 59b3ca2ecc..d395715422 100644
--- a/python/sdist/amici/jax/jax.template.py
+++ b/python/sdist/amici/jax/jax.template.py
@@ -101,7 +101,7 @@ def state_ids(self):
 
     @property
     def parameter_ids(self):
-        return TPL_PK_IDS
+        return TPL_P_IDS
 
 
 Model = JAXModel_TPL_MODEL_NAME