CI: install with uv; consume jet-array 0.1.0a2 from PyPI

.github/workflows/test.yml

@@ -1,5 +1,7 @@
-# Requires jet-array to be installable from PyPI.
-# Publish jet-array first (tag v0.1.0 on thisiscam/jet-array → CI auto-publishes).
+# Installs with uv so the [tool.uv] overrides apply: the oryx/jet-array git
+# sources, the dropped stable tensorflow-probability, and the tfp-nightly pin
+# that keeps `jax.interpreters.xla.pytype_aval_mappings` available. Plain pip
+# ignores those blocks and the import crashes, so do NOT switch back to pip.
 name: tests
 
 on:
@@ -14,16 +16,14 @@ jobs:
     strategy:
       fail-fast: false
       matrix:
-        python-version: ["3.10", "3.11", "3.12"]
+        python-version: ["3.11", "3.12"]  # pyproject requires-python = ">=3.11"
     steps:
       - uses: actions/checkout@v4
-      - uses: actions/setup-python@v5
+      - uses: astral-sh/setup-uv@v5
         with:
           python-version: ${{ matrix.python-version }}
-          cache: pip
+          enable-cache: true
       - name: Install
-        run: |
-          python -m pip install --upgrade pip
-          pip install -e ".[test]"
+        run: uv sync --extra test --locked
       - name: Run tests
-        run: pytest -v
+        run: uv run --extra test python -m pytest -v

pyproject.toml

@@ -25,14 +25,13 @@ classifiers = [
     "Intended Audience :: Science/Research",
     "Operating System :: OS Independent",
     "Programming Language :: Python :: 3",
-    "Programming Language :: Python :: 3.10",
     "Programming Language :: Python :: 3.11",
     "Programming Language :: Python :: 3.12",
     "Topic :: Scientific/Engineering :: Mathematics",
 ]
 dependencies = [
     "jax>=0.8,<0.9",
-    "jet-array==0.1.0a1",
+    "jet-array==0.1.0a2",
     "jaxopt>=0.8",
     "optimistix>=0.0.6",
     "quadax>=0.2.9",
@@ -87,12 +86,10 @@ constraint-dependencies = [
 # upstream fix lands.
 [tool.uv.sources]
 oryx = { git = "https://github.com/thisiscam/oryx.git", branch = "copulaad_support" }
-# PyPI's jet-array==0.1.0a1 predates the log-space Taylor expansion
-# (sign, log|c|) representation that `bell.py`'s `log_jet=True` path
-# uses, and also predates the log_add NaN-gradient shift-guard at
-# structural zeros.  Pin to the public fork until a new wheel is
-# released.
-jet-array = { git = "https://github.com/thisiscam/jet-array.git", branch = "main" }
+# jet-array 0.1.0a2 (PyPI) ships the log-space Taylor expansion (sign, log|c|)
+# that bell.py's log_jet=True path uses, plus the pow_p/scan_p jet rules and the
+# log_add structural-zero NaN-gradient guard — so the published wheel now
+# suffices and no git source is needed.
 
 [tool.hatch.build.targets.wheel]
 packages = ["acopula"]