Skip to content

Commit

Permalink
ci: Use --find-links to install jax and jaxlib on Python 3.7 (#2422)
Browse files Browse the repository at this point in the history
* As Python 3.7 versions of jax and jaxlib are no longer hosted on PyPI, use
  the --find-links option and the Google archive of jax release wheels to
  install the dependencies.
  • Loading branch information
matthewfeickert committed Jan 5, 2024
1 parent 064cdf9 commit 51e55f7
Showing 1 changed file with 7 additions and 0 deletions.
7 changes: 7 additions & 0 deletions .github/workflows/ci.yml
Original file line number Diff line number Diff line change
Expand Up @@ -40,7 +40,14 @@ jobs:
with:
python-version: ${{ matrix.python-version }}

- name: Install dependencies (Python 3.7)
if: matrix.python-version == '3.7'
run: |
python -m pip install --upgrade pip setuptools wheel
python -m pip install --find-links https://storage.googleapis.com/jax-releases/jax_releases.html --upgrade .[test]
- name: Install dependencies
if: matrix.python-version != '3.7'
run: |
python -m pip install --upgrade pip setuptools wheel
python -m pip install --upgrade .[test]
Expand Down

0 comments on commit 51e55f7

Please sign in to comment.