fix: pin orbax-checkpoint version to 0.11.23 (#11)

cmj2002 · web-flow · commit 414d09035b5c · 2025-10-19T01:20:11.000-04:00
diff --git a/setup.py b/setup.py
@@ -17,6 +17,7 @@ def get_base_requirements():
     return [
         "jax[cuda12]==0.5.3",
         "flax==0.10.5",
+        "orbax-checkpoint==0.11.23",
         "gymnasium",
         "gymnasium_robotics",
         'shimmy==1.3.0',
