Skip to content

Commit

Permalink
fix jax version
Browse files Browse the repository at this point in the history
  • Loading branch information
rdyro committed Dec 2, 2024
1 parent 07dd079 commit c3ffffd
Showing 1 changed file with 3 additions and 5 deletions.
8 changes: 3 additions & 5 deletions .github/workflows/ci-build.yml
Original file line number Diff line number Diff line change
@@ -39,7 +39,7 @@ jobs:
strategy:
matrix:
python-version: ["3.11", "3.12"]
jax-version: ["0.4.30", "0.4.35", "latest"]
jax-version: ["0.4.33", "0.4.35", "latest"]
runs-on: ubuntu-latest
timeout-minutes: 5
steps:
@@ -69,12 +69,10 @@ jobs:
source ~/.venv/bin/activate
uv pip install pytest && pip install pytest
if [[ ${{ matrix.jax-version }} == "latest" ]]; then
uv pip install -U jax
uv pip install -U torch jax .
else
uv pip install jax==${{ matrix.jax-version }}
uv pip install torch jax==${{ matrix.jax-version }} .
fi
uv pip install -U torch
uv pip install -U .
- name: Get extension path
id: cpp-extension-cache
run: |

0 comments on commit c3ffffd

Please sign in to comment.