Replace nox with tox + tox-uv for multi-version testing

Use uv-venv-lock-runner with PEP 735 dependency groups for factor-based
backend selection (torch/jax/all). Factor prefix syntax in tox.ini
cleanly maps env names like 3.12-torch to the right groups.

Author: acecchini

.github/workflows/ci.yml

@@ -15,38 +15,24 @@ jobs:
     steps:
       - uses: actions/checkout@v4
       - uses: astral-sh/setup-uv@v5
-      - run: uv sync --group dev
-      - run: uv run ruff check src/ tests/
-      - run: uv run ruff format --check src/ tests/
+      - run: uv run tox -e lint
 
   typecheck:
     runs-on: ubuntu-latest
     steps:
       - uses: actions/checkout@v4
       - uses: astral-sh/setup-uv@v5
-      - run: uv sync --group dev
-      - run: uv run pyright src/
+      - run: uv run tox -e typecheck
 
   test:
     runs-on: ubuntu-latest
     strategy:
       fail-fast: false
       matrix:
-        python-version: ["3.12", "3.13", "3.14"]
-        backend: ["numpy-only", "torch", "jax", "all"]
-    name: test (py${{ matrix.python-version }}, ${{ matrix.backend }})
+        python: ["3.12", "3.13", "3.14"]
+        suffix: ["", "-torch", "-jax", "-all"]
+    name: test (py${{ matrix.python }}${{ matrix.suffix }})
     steps:
       - uses: actions/checkout@v4
       - uses: astral-sh/setup-uv@v5
-        with:
-          python-version: ${{ matrix.python-version }}
-      - name: Install core + dev deps
-        run: uv sync --group dev
-      - name: Install torch
-        if: matrix.backend == 'torch' || matrix.backend == 'all'
-        run: uv sync --group dev --group torch
-      - name: Install jax
-        if: matrix.backend == 'jax' || matrix.backend == 'all'
-        run: uv sync --group dev --group jax
-      - name: Run tests
-        run: uv run pytest tests/ -v
+      - run: uv run tox -e "${{ matrix.python }}${{ matrix.suffix }}"

noxfile.py

@@ -15,7 +15,8 @@ dev = [
   "pyright>=1.1.408",
   "pytest>=9.0.2",
   "pre-commit>=4.5.1",
-  "nox>=2025.5.1",
+  "tox>=4.19",
+  "tox-uv>=1",
   "ipykernel>=7.1.0",
   "matplotlib>=3.10.8",
 ]

pyproject.toml

@@ -25,6 +25,9 @@ select = [
   "E113",
   "E115",
   "E117",
+  "E225",
+  "E227",
+  "E228",
 ]
 
 [format]

ruff.toml

@@ -0,0 +1,36 @@
+[tox]
+requires =
+    tox-uv>=1
+env_list =
+    3.12, 3.13, 3.14
+    3.12-torch, 3.13-torch, 3.14-torch
+    3.12-jax, 3.13-jax, 3.14-jax
+    3.12-all, 3.13-all, 3.14-all
+    lint
+    typecheck
+
+[testenv]
+runner = uv-venv-lock-runner
+description = Run tests under {base_python}
+dependency_groups =
+    dev
+    torch: torch
+    jax: jax
+    all: torch
+    all: jax
+commands =
+    pytest {posargs:tests/} -v
+
+[testenv:lint]
+description = Run ruff linter and formatter checks
+skip_install = true
+dependency_groups = dev
+commands =
+    ruff check src/ tests/
+    ruff format --check src/ tests/
+
+[testenv:typecheck]
+description = Run pyright type checker
+dependency_groups = dev
+commands =
+    pyright src/