From 37715aa08bd711c194d62074e81437ae1a9c34c0 Mon Sep 17 00:00:00 2001 From: Luke Baumann Date: Fri, 19 Sep 2025 20:44:01 +0000 Subject: [PATCH] Update GitHub action unittest matrix Adds a JAX 0.7.1, 0.7.2, and nightly version and Python 3.13 to the version lists. --- .github/workflows/test.yml | 17 +++++++++++++---- 1 file changed, 13 insertions(+), 4 deletions(-) diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml index 1830deb..5b18802 100644 --- a/.github/workflows/test.yml +++ b/.github/workflows/test.yml @@ -23,12 +23,18 @@ jobs: strategy: fail-fast: false matrix: - python-version: ['3.10', '3.11', '3.12'] - jax-version: ['0.5.3', '0.6.0', '0.6.1', '0.6.2', '0.7.0'] + python-version: ['3.10', '3.11', '3.12', '3.13'] + jax-version: ['0.5.3', '0.6.0', '0.6.1', '0.6.2', '0.7.0', '0.7.1', '0.7.2', 'nightly'] exclude: - python-version: '3.10' jax-version: '0.7.0' + - python-version: '3.10' + jax-version: '0.7.1' + - python-version: '3.10' + jax-version: '0.7.2' + - python-version: '3.10' + jax-version: 'nightly' runs-on: ubuntu-latest timeout-minutes: 30 @@ -40,11 +46,14 @@ jobs: - uses: actions/checkout@v4 # Install deps - - uses: actions/setup-python@v5 + - uses: actions/setup-python@v6 with: python-version: ${{ matrix.python-version }} - run: pip --version - - run: pip install -e . "jax==${{ matrix.jax-version }}" + - if: ${{ matrix.jax-version == 'nightly' }} + run: pip install -e . -U --pre jax jaxlib -i https://us-python.pkg.dev/ml-oss-artifacts-published/jax/simple/ + - if: ${{ matrix.jax-version != 'nightly' }} + run: pip install -e . "jax==${{ matrix.jax-version }}" - run: pip freeze # Run tests