From e9a226b7c6d570046d9ff9b278bba7aeba1c15ee Mon Sep 17 00:00:00 2001
From: Branwen Snelling <branwen.snelling@crick.ac.uk>
Date: Mon, 2 Mar 2026 12:09:49 +0000
Subject: [PATCH] rm interpax dep and jax pin to update lineax

---
 python/sdist/amici/exporters/jax/jax.template.py | 1 -
 python/sdist/pyproject.toml                      | 3 +--
 2 files changed, 1 insertion(+), 3 deletions(-)

diff --git a/python/sdist/amici/exporters/jax/jax.template.py b/python/sdist/amici/exporters/jax/jax.template.py
index 37b6339d5a..4b74ae2702 100644
--- a/python/sdist/amici/exporters/jax/jax.template.py
+++ b/python/sdist/amici/exporters/jax/jax.template.py
@@ -5,7 +5,6 @@
 import jax.numpy as jnp
 import jax.random as jr  # noqa: F401
 import jaxtyping as jt  # noqa: F401
-from interpax import interp1d  # noqa: F401
 from jax.numpy import inf as oo  # noqa: F401
 from jax.numpy import nan as nan  # noqa: F401
 
diff --git a/python/sdist/pyproject.toml b/python/sdist/pyproject.toml
index 425ce7d533..334d2f51ab 100644
--- a/python/sdist/pyproject.toml
+++ b/python/sdist/pyproject.toml
@@ -87,12 +87,11 @@ examples = [
     "scipy",
 ]
 jax = [
-    "jax>=0.7.2,<0.8.2",
+    "jax>=0.7.2",
     "diffrax>=0.7.0",
     "jaxtyping>=0.2.34",
     "equinox>=0.13.2",
     "optimistix>=0.0.9",
-    "interpax>=0.3.9",
 ]
 sciml = [
     "h5py"