Skip to content

[CI] Add old Jax tests #4193

[CI] Add old Jax tests

[CI] Add old Jax tests #4193

Workflow file for this run

name: ci
on:
push:
branches:
- main
pull_request:
concurrency:
group: ci-${{ github.event.pull_request.number || github.sha }}
cancel-in-progress: true
jobs:
test:
runs-on: "${{ matrix.os }}"
strategy:
matrix:
python-version: ["3.9", "3.12"]
os: [ubuntu-latest]
steps:
- uses: actions/checkout@v2
with:
submodules: recursive
- uses: actions/setup-python@v2
with:
python-version: "${{ matrix.python-version }}"
architecture: 'x64'
- name: install
run: |
if [ "${{ matrix.python-version }}" = "3.9" ]; then
pip install "numpy<2.0.0" "jax==0.4.6" "jaxlib==0.4.6"
else
pip install jax jaxlib
fi
make install
- name: test
run: |
make install-dev
make test-with-codecov
- name: Upload coverage reports to Codecov
uses: codecov/codecov-action@v3
env:
CODECOV_TOKEN: ${{ secrets.CODECOV_TOKEN }}