Update pyproject dependencies.

thebuckleylab · Dec 5, 2024 · ac463c3 · ac463c3
1 parent 253b272
commit ac463c3
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 5 deletions.
diff --git a/README.md b/README.md
@@ -46,8 +46,9 @@ pip install jpc
 ```
 
 Requires Python 3.9+, JAX 0.4.23+, [Equinox](https://github.com/patrick-kidger/equinox) 
-0.11.2+, [Diffrax](https://github.com/patrick-kidger/diffrax) 0.6.0+, and 
-[Optax](https://github.com/google-deepmind/optax) 0.2.4+. 
+0.11.2+, [Diffrax](https://github.com/patrick-kidger/diffrax) 0.6.0+, 
+[Optax](https://github.com/google-deepmind/optax) 0.2.4+, and
+[jaxtyping](https://github.com/patrick-kidger/jaxtyping) 0.2.24+.
 
 For GPU usage, upgrade jax to the appropriate cuda version (12 as an example 
 here).

diff --git a/docs/index.md b/docs/index.md
@@ -30,8 +30,9 @@ pip install jpc
 ```
 
 Requires Python 3.9+, JAX 0.4.23+, [Equinox](https://github.com/patrick-kidger/equinox) 
-0.11.2+, [Diffrax](https://github.com/patrick-kidger/diffrax) 0.6.0+, and 
-[Optax](https://github.com/google-deepmind/optax) 0.2.4+. 
+0.11.2+, [Diffrax](https://github.com/patrick-kidger/diffrax) 0.6.0+, 
+[Optax](https://github.com/google-deepmind/optax) 0.2.4+, and
+[jaxtyping](https://github.com/patrick-kidger/jaxtyping) 0.2.24+.
 
 For GPU usage, upgrade jax to the appropriate cuda version (12 as an example 
 here).

diff --git a/pyproject.toml b/pyproject.toml
@@ -35,7 +35,7 @@ dependencies = [
     "jax>=0.4.23",
     "equinox>=0.11.2",
     "diffrax>=0.6.0",
-    "optax>=0.2.2",
+    "optax>=0.2.4",
     "jaxtyping>=0.2.24"
 ]