From 78df36f2f0f173ad651b81527cda8d55f85028b0 Mon Sep 17 00:00:00 2001
From: init-22 <isaac.patole022@gmail.com>
Date: Fri, 15 Nov 2024 00:42:26 +0530
Subject: [PATCH] fix: tf version to use numpy < 2

---
 docker/Dockerfile | 2 --
 setup.cfg         | 4 ++--
 2 files changed, 2 insertions(+), 4 deletions(-)

diff --git a/docker/Dockerfile b/docker/Dockerfile
index 497ffb2c1..88fc55243 100644
--- a/docker/Dockerfile
+++ b/docker/Dockerfile
@@ -87,8 +87,6 @@ RUN if [ "$framework" = "jax" ] ; then \
 
 RUN cd /algorithmic-efficiency && pip install -e '.[full]'
 
-RUN cd /algorithmic-efficiency && pip install -e '.[wandb]'
-
 RUN cd /algorithmic-efficiency && git fetch origin 
 RUN cd /algorithmic-efficiency && git pull
 
diff --git a/setup.cfg b/setup.cfg
index a74faa197..2a300469a 100644
--- a/setup.cfg
+++ b/setup.cfg
@@ -41,7 +41,7 @@ install_requires =
   docker==7.1.0
   numpy>=1.26.4
   pandas==2.2.3
-  tensorflow==2.18.0
+  tensorflow==2.17.0
   tensorflow-datasets==4.9.7
   tensorflow-addons==0.23.0
   gputil==1.4.0
@@ -105,7 +105,7 @@ librispeech_conformer =
 
 wmt =
   sentencepiece==0.2.0
-  tensorflow-text==2.18.0
+  tensorflow-text==2.17.0
   sacrebleu==2.4.3
 
 # Frameworks #