From 316c5b7768485f444bcaa95774a7efa7bf74c802 Mon Sep 17 00:00:00 2001
From: Patrick Kidger <33688385+patrick-kidger@users.noreply.github.com>
Date: Fri, 8 Apr 2022 10:53:08 +0100
Subject: [PATCH] Fixed no jaxlib in workflows.

---
 .github/workflows/release.yml   | 2 +-
 .github/workflows/run_tests.yml | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/.github/workflows/release.yml b/.github/workflows/release.yml
index d7577378..c7fa88ad 100644
--- 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
index 5039a346..30305c8a 100644
--- 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/action@v2.0.3