Pin flax

brentyi · Jul 27, 2023 · 7a01d94 · 7a01d94
1 parent a45e4d7
commit 7a01d94
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -43,8 +43,7 @@ dev = [
     "pyright>=1.1.264",
     "mypy>=0.991",
     "numpy>=1.20.0",
-    "orbax==0.1.6",  # Needed for flax install in Python 3.7 as of 7/27/2023.
-    "flax>=0.6.0",
+    "flax==0.6.8",  # Needed for Python 3.7 compatibility.
     "pydantic>=1.10.2",
     "coverage[toml]>=6.5.0"
 ]