From 5d48fb3465678dd6e47249ad49164e6fa6000457 Mon Sep 17 00:00:00 2001
From: Sam-Armstrong <samuel_e_armstrong@yahoo.co.uk>
Date: Mon, 22 Jul 2024 23:25:17 +0100
Subject: [PATCH] fix: update requirement_mappings for tf unpin

---
 docker/requirement_mappings.json | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/docker/requirement_mappings.json b/docker/requirement_mappings.json
index 09d17a4874299..e87d8e8b671e6 100644
--- 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"],