install jaxlib manually for test

sail-sg · Nov 30, 2023 · d68382a · d68382a
1 parent fc852e1
commit d68382a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
@@ -17,4 +17,4 @@ jobs:
         python-version: 3.11
     - name: Test
       run: |
-        pip install absl-py graphviz && pip install . && python3 -m unittest discover -p "*_test.py"
+        pip install absl-py graphviz jax jaxlib && pip install . && python3 -m unittest discover -p "*_test.py"