fix: also pinned the version in the requirement_mappings.json

imsoumya18 · Mar 14, 2024 · 8d8f86a · 8d8f86a
1 parent fe3c47e
commit 8d8f86a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/docker/requirement_mappings.json b/docker/requirement_mappings.json
@@ -1,5 +1,5 @@
 {
-  "tensorflow": ["tensorflow-cpu", "tensorflow-probability"],
+  "tensorflow": ["tensorflow-cpu==2.15.1", "tensorflow-probability"],
   "jax": ["ml-dtypes","jax[cpu]","dm-haiku", "flax", "jaxlib"],
   "numpy": ["numpy"],
   "paddle": ["paddlepaddle"],