Fixes (#1)

* Fixes * Black * Test fix * Test fix * workflow fixes
2026-09-09 11:24:55 +08:00 · 2022-07-11 13:03:39 +01:00
parent 207faaabfd
commit b5c25822ff
8 changed files with 29 additions and 7 deletions
@@ -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 }}
@@ -42,7 +42,7 @@ jobs:
      - name: Install dependencies
        run: |
          python -m pip install --upgrade pip
-          python -m pip install pytest wheel jaxlib
+          python -m pip install pytest wheel beartype equinox jaxlib

      - name: Checks with pre-commit
        uses: pre-commit/action@v2.0.3