fix(pyproject): update flax

Signed-off-by: Cameron Smith <cameron.ray.smith@gmail.com>
pinellolab · Feb 28, 2024 · 6b664bf · 6b664bf
1 parent 3782f01
commit 6b664bf
Show file tree

Hide file tree

Showing 2 changed files with 122 additions and 11 deletions.
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -34,7 +34,7 @@ beartype = { version = "0.16.4", source = "pypi" }
 chex = "0.1.85"
 click = "8.1.7"
 colorlog = "6.7.0"
-flax = "0.6.1"
+flax = "0.8.1"
 h5py = "3.8.0"
 jax = "0.4.23"
 jaxlib = "0.4.23"