From 19b6c99edfa5e9211b81609c4e29d3fdc963dde8 Mon Sep 17 00:00:00 2001
From: Alexandru Fikl <alexfikl@gmail.com>
Date: Sun, 5 Jul 2026 10:36:14 +0300
Subject: [PATCH] feat: blacklist jax 0.10.2

---
 .gitlab-ci.yml          | 2 +-
 .test-conda-env-py3.yml | 1 +
 pyproject.toml          | 2 +-
 3 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/.gitlab-ci.yml b/.gitlab-ci.yml
index e40c8e74..5d2a4670 100644
--- a/.gitlab-ci.yml
+++ b/.gitlab-ci.yml
@@ -71,7 +71,7 @@ Python 3 Conda:
     export EXTRA_INSTALL="scipy"
 
     # Avoid crashes like https://gitlab.tiker.net/inducer/arraycontext/-/jobs/536021
-    sed -i 's/jax/jax !=0.4.6/' .test-conda-env-py3.yml
+    sed -i '/^- jax$/s/$/ !=0.4.6/' .test-conda-env-py3.yml
 
     curl -L -O https://gitlab.tiker.net/inducer/ci-support/raw/main/build-and-test-py-project-within-miniconda.sh
     . ./build-and-test-py-project-within-miniconda.sh
diff --git a/.test-conda-env-py3.yml b/.test-conda-env-py3.yml
index 66daddad..0487765b 100644
--- a/.test-conda-env-py3.yml
+++ b/.test-conda-env-py3.yml
@@ -15,3 +15,4 @@ dependencies:
 - islpy
 - pip
 - jax
+- jaxlib !=0.10.2
diff --git a/pyproject.toml b/pyproject.toml
index 850e3536..32d3a544 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -36,7 +36,7 @@ dependencies = [
 
 [project.optional-dependencies]
 jax = [
-    "jax>=0.4",
+    "jax>=0.4,!=0.10.2",
 ]
 pyopencl = [
     "islpy>=2024.1",