From a514debd75b680d75278bf0c6ed80a5813a759a8 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Fabian=20Fr=C3=B6hlich?= <fabian@schaluck.com>
Date: Sat, 9 Nov 2024 23:16:12 +0000
Subject: [PATCH] update template

---
 .pre-commit-config.yaml            |  6 ------
 python/sdist/amici/jax.template.py | 16 ++++++++++++++++
 2 files changed, 16 insertions(+), 6 deletions(-)

diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
index f16458b29a..26395d1a0a 100644
--- a/.pre-commit-config.yaml
+++ b/.pre-commit-config.yaml
@@ -27,10 +27,4 @@ repos:
         - --config
         - python/sdist/pyproject.toml
 
--   repo: https://github.com/asottile/pyupgrade
-    rev: v3.17.0
-    hooks:
-    -   id: pyupgrade
-        args: ["--py310-plus"]
-
 exclude: '^(ThirdParty|models)/'
diff --git a/python/sdist/amici/jax.template.py b/python/sdist/amici/jax.template.py
index c52f29a78f..b6048b57f5 100644
--- a/python/sdist/amici/jax.template.py
+++ b/python/sdist/amici/jax.template.py
@@ -106,3 +106,19 @@ def Jy(y, my, sigmay):
 TPL_JY_EQ
 
         return TPL_JY_RET
+
+    @property
+    def parameter_ids(self):
+        return TPL_P_IDS
+
+    @property
+    def fixed_parameter_ids(self):
+        return TPL_K_IDS
+
+    @property
+    def observable_ids(self):
+        return TPL_Y_IDS
+
+    @property
+    def state_ids(self):
+        return TPL_X_IDS