update CONTRIBUTING.md

patrick-kidger · Jul 31, 2024 · 317cc9e · 317cc9e
1 parent 2ae053b
commit 317cc9e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/CONTRIBUTING.md b/CONTRIBUTING.md
@@ -28,7 +28,7 @@ Now make your changes. Make sure to include additional tests if necessary.
 Next verify the tests all pass:
 
 ```bash
-pip install pytest cloudpickle
+pip install -r test/requirements.txt
 pip install torch --extra-index-url https://download.pytorch.org/whl/cpu
 pytest
 ```

diff --git a/test/requirements.txt b/test/requirements.txt
@@ -2,7 +2,7 @@ beartype
 cloudpickle
 equinox
 IPython
-jaxlib
+jax
 pytest
 pytest-asyncio
 tensorflow