fix: update requirement_mappings for tf unpin

ivy-llc · Jul 22, 2024 · 5d48fb3 · 5d48fb3
1 parent 6ddd644
commit 5d48fb3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/docker/requirement_mappings.json b/docker/requirement_mappings.json
@@ -1,6 +1,6 @@
 {
-  "tensorflow": ["tensorflow-cpu==2.15.1", "tensorflow-probability"],
-  "jax": ["ml-dtypes","jax[cpu]","dm-haiku", "flax", "jaxlib"],
+  "tensorflow": ["tensorflow", "tensorflow-probability"],
+  "jax": ["ml-dtypes", "jax[cpu]", "dm-haiku", "flax", "jaxlib"],
   "numpy": ["numpy"],
   "paddle": ["paddlepaddle"],
   "mxnet": ["mxnet"],