Updating the right dependency

parambole · parambole · commit 3831423e1b14 · 2025-04-25T23:40:41.000Z
diff --git a/requirements.txt b/requirements.txt
@@ -19,12 +19,12 @@ Pillow
 pylint
 pyink
 pytest==8.2.2
-tensorflow==2.17.0
+tensorflow>=2.17.0
 tensorflow-datasets>=4.9.6
 ruff>=0.1.5,<=0.2
 git+https://github.com/mlperf/logging.git
 opencv-python-headless==4.10.0.84
-orbax-checkpoint>=0.10.3
+orbax-checkpoint==0.10.3
 tokenizers==0.21.0
 huggingface_hub==0.30.2
 transformers==4.48.1
diff --git a/requirements_with_jax_stable_stack.txt b/requirements_with_jax_stable_stack.txt
@@ -14,7 +14,7 @@ jaxlib>=0.4.30
 Jinja2
 opencv-python-headless==4.10.0.84
 optax>=0.2.3
-orbax-checkpoint>=0.10.3
+orbax-checkpoint==0.10.3
 parameterized
 Pillow
 pyink
