Pinning versions of libraries used in tests

PiperOrigin-RevId: 718010530
google-deepmind · Jan 21, 2025 · 102f0c4 · 102f0c4
1 parent 9182734
commit 102f0c4
Showing 1 changed file with 10 additions and 10 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -62,15 +62,15 @@ 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==2.1.0",
+  "immutabledict==4.2.1",
+  "numpy==2.2.1",
+  "distrax==0.1.5",
+  "jax==0.4.38",
+  "jaxlib==0.4.38",
+  "dm-haiku==0.0.13",
+  "dm-tree==0.1.8",
+  "optax>=0.2.4",
 ]