Fix dependencies.

PiperOrigin-RevId: 542534464 Change-Id: I4709b260d899edeeb46a54b2053f8a237399d312
google-deepmind · Jun 27, 2023 · 8017d26 · 8017d26
1 parent 51bd6c4
commit 8017d26
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 5 deletions.
diff --git a/README.md b/README.md
@@ -89,7 +89,7 @@ Please see the [JAX documentation](https://github.com/google/jax#installation) f
 Before running any code, make sure to activate the conda environment and set the `PYTHONPATH`:
 ```bash
 conda activate nnch
-export PYTHONPATH=$(pwd)
+export PYTHONPATH=$(pwd)/..
 ```
 
 We provide an example of a training and evaluation run at:

diff --git a/requirements.txt b/requirements.txt
@@ -1,8 +1,9 @@
-jax
+absl-py
 dm-haiku
+dm-tree
+git+https://github.com/deepmind/einshape
+jax
 numpy
 optax
 tqdm
-git+https://github.com/deepmind/einshape
-dm-tree
-absl-py
+typing-extensions