fix: tf version to use numpy < 2

mlcommons · Nov 14, 2024 · 78df36f · 78df36f
1 parent 67b9f15
commit 78df36f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git 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
@@ -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 #