update template

.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)/'

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