From 92002d62f5a19c320b8d25c19b1ea81bb8c7f624 Mon Sep 17 00:00:00 2001 From: Samuel Marks <807580+SamuelMarks@users.noreply.github.com> Date: Mon, 3 Nov 2025 18:57:07 -0800 Subject: [PATCH] =?UTF-8?q?[dependencies/dockerfiles/maxtext=5F{db=5Fdepen?= =?UTF-8?q?dencies,dependencies,gpu=5Fdependencies,jax=5Fai=5Fimage}.Docke?= =?UTF-8?q?rfile]=20`extra=5Fdeps=5Ffrom=5Fgithub.txt`=20is=20now=20only?= =?UTF-8?q?=20searched=20for=20in=20filesystem=20path=20only=E2=80=A6=20so?= =?UTF-8?q?=20`COPY`=20over=20there?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- dependencies/dockerfiles/maxtext_db_dependencies.Dockerfile | 6 +++--- dependencies/dockerfiles/maxtext_dependencies.Dockerfile | 6 +++--- .../dockerfiles/maxtext_gpu_dependencies.Dockerfile | 6 +++--- dependencies/dockerfiles/maxtext_jax_ai_image.Dockerfile | 6 +++--- 4 files changed, 12 insertions(+), 12 deletions(-) diff --git a/dependencies/dockerfiles/maxtext_db_dependencies.Dockerfile b/dependencies/dockerfiles/maxtext_db_dependencies.Dockerfile index e5a767e7c..1f7a72aa6 100644 --- a/dependencies/dockerfiles/maxtext_db_dependencies.Dockerfile +++ b/dependencies/dockerfiles/maxtext_db_dependencies.Dockerfile @@ -40,9 +40,9 @@ ENV MAXTEXT_REPO_ROOT=/deps WORKDIR /deps # Copy setup files and dependency files separately for better caching -COPY tools/setup /deps/tools/setup/ -COPY dependencies/requirements/ /deps/dependencies/requirements/ -COPY src/install_maxtext_extra_deps/extra_deps_from_github.txt /deps/dependencies/requirements/ +COPY tools/setup tools/setup/ +COPY dependencies/requirements/ dependencies/requirements/ +COPY src/install_maxtext_extra_deps/extra_deps_from_github.txt src/install_maxtext_extra_deps/ # Install dependencies - these steps are cached unless the copied files change RUN echo "Running command: bash setup.sh MODE=$ENV_MODE JAX_VERSION=$ENV_JAX_VERSION LIBTPU_GCS_PATH=${ENV_LIBTPU_GCS_PATH} DEVICE=${ENV_DEVICE}" diff --git a/dependencies/dockerfiles/maxtext_dependencies.Dockerfile b/dependencies/dockerfiles/maxtext_dependencies.Dockerfile index 4b86c08ff..aa133f80f 100644 --- a/dependencies/dockerfiles/maxtext_dependencies.Dockerfile +++ b/dependencies/dockerfiles/maxtext_dependencies.Dockerfile @@ -40,9 +40,9 @@ ENV MAXTEXT_REPO_ROOT=/deps WORKDIR /deps # Copy setup files and dependency files separately for better caching -COPY tools/setup /deps/tools/setup/ -COPY dependencies/requirements/ /deps/dependencies/requirements/ -COPY src/install_maxtext_extra_deps/extra_deps_from_github.txt /deps/dependencies/requirements/ +COPY tools/setup tools/setup/ +COPY dependencies/requirements/ dependencies/requirements/ +COPY src/install_maxtext_extra_deps/extra_deps_from_github.txt src/install_maxtext_extra_deps/ # Install dependencies - these steps are cached unless the copied files change RUN echo "Running command: bash setup.sh MODE=$ENV_MODE JAX_VERSION=$ENV_JAX_VERSION LIBTPU_GCS_PATH=${ENV_LIBTPU_GCS_PATH} DEVICE=${ENV_DEVICE}" diff --git a/dependencies/dockerfiles/maxtext_gpu_dependencies.Dockerfile b/dependencies/dockerfiles/maxtext_gpu_dependencies.Dockerfile index cba712581..b8a3605b0 100644 --- a/dependencies/dockerfiles/maxtext_gpu_dependencies.Dockerfile +++ b/dependencies/dockerfiles/maxtext_gpu_dependencies.Dockerfile @@ -42,9 +42,9 @@ ENV MAXTEXT_REPO_ROOT=/deps WORKDIR /deps # Copy setup files and dependency files separately for better caching -COPY tools/setup /deps/tools/setup/ -COPY dependencies/requirements/ /deps/dependencies/requirements/ -COPY src/install_maxtext_extra_deps/extra_deps_from_github.txt /deps/dependencies/requirements/ +COPY tools/setup tools/setup/ +COPY dependencies/requirements/ dependencies/requirements/ +COPY src/install_maxtext_extra_deps/extra_deps_from_github.txt src/install_maxtext_extra_deps/ # Install dependencies - these steps are cached unless the copied files change RUN echo "Running command: bash setup.sh MODE=$ENV_MODE JAX_VERSION=$ENV_JAX_VERSION DEVICE=${ENV_DEVICE}" diff --git a/dependencies/dockerfiles/maxtext_jax_ai_image.Dockerfile b/dependencies/dockerfiles/maxtext_jax_ai_image.Dockerfile index edcadaf9c..6dcc6a054 100644 --- a/dependencies/dockerfiles/maxtext_jax_ai_image.Dockerfile +++ b/dependencies/dockerfiles/maxtext_jax_ai_image.Dockerfile @@ -16,9 +16,9 @@ ENV MAXTEXT_REPO_ROOT=/deps WORKDIR /deps # Copy setup files and dependency files separately for better caching -COPY tools/setup /deps/tools/setup/ -COPY dependencies/requirements/ /deps/dependencies/requirements/ -COPY src/install_maxtext_extra_deps/extra_deps_from_github.txt /deps/dependencies/requirements/ +COPY tools/setup tools/setup/ +COPY dependencies/requirements/ dependencies/requirements/ +COPY src/install_maxtext_extra_deps/extra_deps_from_github.txt src/install_maxtext_extra_deps/ # For JAX AI tpu training images 0.4.37 AND 0.4.35 # Orbax checkpoint installs the latest version of JAX,