Fixed no jaxlib in workflows.

patrick-kidger · Apr 8, 2022 · 316c5b7 · 316c5b7
1 parent f3b55d5
commit 316c5b7
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/.github/workflows/release.yml b/.github/workflows/release.yml
@@ -14,7 +14,7 @@ jobs:
         with:
             python-version: "3.8"
             test-script: |
-                python -m pip install pytest
+                python -m pip install pytest jaxlib
                 cp -r ${{ github.workspace }}/tests ./tests
                 pytest
             pypi-token: ${{ secrets.pypi_token }}

diff --git a/.github/workflows/run_tests.yml b/.github/workflows/run_tests.yml
@@ -25,7 +25,7 @@ jobs:
       - name: Install dependencies
         run: |
           python -m pip install --upgrade pip
-          python -m pip install pytest wheel optax
+          python -m pip install pytest wheel optax jaxlib
 
       - name: Checks with pre-commit
         uses: pre-commit/[email protected]