Move tools to docker image

RUEI4341 · RUEI4341 · commit d6150e0d7f12 · 2026-03-30T17:38:05.000+08:00
diff --git a/src/dependencies/dockerfiles/maxtext_gpu_dependencies.Dockerfile b/src/dependencies/dockerfiles/maxtext_gpu_dependencies.Dockerfile
@@ -66,6 +66,8 @@ COPY ${PACKAGE_DIR}/maxtext/ src/maxtext/
 COPY ${PACKAGE_DIR}/MaxText/ src/MaxText/
 COPY tests*/ tests/
 COPY benchmarks*/ benchmarks/
+COPY tools/setup*/ tools/setup/
+
 
 # Download test assets from GCS if building image with test assets
 ARG INCLUDE_TEST_ASSETS=false
diff --git a/src/dependencies/dockerfiles/maxtext_tpu_dependencies.Dockerfile b/src/dependencies/dockerfiles/maxtext_tpu_dependencies.Dockerfile
@@ -66,6 +66,8 @@ COPY ${PACKAGE_DIR}/maxtext/ src/maxtext/
 COPY ${PACKAGE_DIR}/MaxText/ src/MaxText/
 COPY tests*/ tests/
 COPY benchmarks*/ benchmarks/
+COPY tools/setup*/ tools/setup/
+
 
 # Download test assets from GCS if building image with test assets
 ARG INCLUDE_TEST_ASSETS=false
