From 0695193b5f87f4772d276db620882508f8145af2 Mon Sep 17 00:00:00 2001
From: Patrick Kidger <33688385+patrick-kidger@users.noreply.github.com>
Date: Mon, 11 Jul 2022 12:58:40 +0100
Subject: [PATCH] workflow fixes

---
 .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 75e31fe..ad9c799 100644
--- a/.github/workflows/release.yml
+++ b/.github/workflows/release.yml
@@ -33,7 +33,7 @@ jobs:
         with:
             python-version: "3.8"
             test-script: |
-                python -m pip install pytest jax jaxlib typeguard
+                python -m pip install pytest beartype equinox jaxlib
                 cp -r ${{ github.workspace }}/test ./test
                 pytest
             pypi-token: ${{ secrets.pypi_token }}
diff --git a/.github/workflows/run_tests.yml b/.github/workflows/run_tests.yml
index ef99b24..56d0d23 100644
--- a/.github/workflows/run_tests.yml
+++ b/.github/workflows/run_tests.yml
@@ -42,7 +42,7 @@ jobs:
       - name: Install dependencies
         run: |
           python -m pip install --upgrade pip
-          python -m pip install pytest wheel jaxlib beartype equinox
+          python -m pip install pytest wheel beartype equinox jaxlib
 
       - name: Checks with pre-commit
         uses: pre-commit/action@v2.0.3