feat: blacklist jax 0.10.2

Author: alexfikl

.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

.test-conda-env-py3.yml

@@ -15,3 +15,5 @@ dependencies:
 - islpy
 - pip
 - jax
+# https://github.com/inducer/arraycontext/pull/366
+- jaxlib !=0.10.2

pyproject.toml

@@ -36,7 +36,8 @@ dependencies = [
 
 [project.optional-dependencies]
 jax = [
-    "jax>=0.4",
+    # https://github.com/inducer/arraycontext/pull/366
+    "jax>=0.4,!=0.10.2",
 ]
 pyopencl = [
     "islpy>=2024.1",