Pinning versions of libraries used in tests

PiperOrigin-RevId: 718010530
google-deepmind · Jan 21, 2025 · 626dc01 · 626dc01
1 parent 9182734
commit 626dc01
Showing 1 changed file with 9 additions and 10 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -62,15 +62,14 @@ examples = [
   "tensorflow>=2.8.0",
   "tensorflow_datasets>=4.5.2",
 ]
-tests = [  # these should be version pinned?
+tests = [
   "pytest-xdist",
-  "absl-py>=0.12.0",
-  "immutabledict>=2.2.1",
-  "numpy>=1.22",
-  "distrax>=0.1.3",
-  "jax>=0.4.27",
-  "jaxlib>=0.4.27",
-  "dm-haiku>=0.0.9",
-  "dm-tree>=0.1.7",
-  "optax>=0.1.4",
+  "absl-py==0.12.0",
+  "immutabledict==2.2.1",
+  "numpy==1.23",
+  "distrax==0.1.4",
+  "jax==0.4.27",
+  "jaxlib==0.4.27",
+  "dm-haiku==0.0.9",
+  "dm-tree==0.1.7",
 ]