From 173ddc5a2b90f1e77ee0960f871751f2e7e402d3 Mon Sep 17 00:00:00 2001 From: Vladislav Kozlov Date: Wed, 27 Nov 2024 13:58:49 -0800 Subject: [PATCH] Update maxtext docker --- .github/container/Dockerfile.maxtext | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/.github/container/Dockerfile.maxtext b/.github/container/Dockerfile.maxtext index a38c46240..17bc6f998 100644 --- a/.github/container/Dockerfile.maxtext +++ b/.github/container/Dockerfile.maxtext @@ -57,12 +57,13 @@ RUN echo "tensorflow-text @ file://$(ls /opt/tensorflow_text*.whl)" >> /opt/pip- RUN <<"EOF" bash -ex git-clone.sh ${URLREF_MAXTEXT} ${SRC_PATH_MAXTEXT} echo "-r ${SRC_PATH_MAXTEXT}/requirements.txt" >> /opt/pip-tools.d/requirements-maxtext.in + +# specify some restrictions to speed up the build and +# avoid pip to download and check all available versions of packages for pattern in \ - "s|@git+https://github.com/mlperf/logging.git||g" \ - "s|absl-py|absl-py==2.1.0|g" \ + "s|absl-py|absl-py>=2.1.0|g" \ "s|protobuf==3.20.3|protobuf>=3.19.0|g" \ "s|tensorflow-datasets|tensorflow-datasets>=4.8.0|g" \ - "s|@git+https://github.com/google/pathways-utils.git||g" \ ; do sed -i "${pattern}" ${SRC_PATH_MAXTEXT}/requirements.txt; done