Unpin the Orbax pip version.

PiperOrigin-RevId: 646215075
google · Jun 24, 2024 · f8ec455 · f8ec455
1 parent bda7da1
commit f8ec455
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pyproject.toml b/pyproject.toml
@@ -17,7 +17,7 @@ dependencies = [
     "jax>=0.4.19",
     "msgpack",
     "optax",
-    "orbax-checkpoint==0.5.15",  # remove this when they release > 0.5.17
+    "orbax-checkpoint",
     "tensorstore",
     "rich>=11.1",
     "typing_extensions>=4.2",