fix orbax version

thorben-frank · Aug 5, 2024 · a3854a2 · a3854a2
1 parent 5df2c91
commit a3854a2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -13,7 +13,7 @@
         "jaxopt",
         "jraph",
         "optax",
-        "orbax-checkpoint",
+        "orbax-checkpoint == 0.5.23",
         "portpicker",
         # 'tensorflow',
         "scikit-learn",