diff --git a/.github/workflows/BuildDockerImages.yml b/.github/workflows/BuildDockerImages.yml index 8049b5b11..9a4154735 100644 --- a/.github/workflows/BuildDockerImages.yml +++ b/.github/workflows/BuildDockerImages.yml @@ -35,6 +35,9 @@ jobs: runs-on: ["self-hosted", "tpu", "${{ matrix.device-type }}"] steps: - uses: actions/checkout@v3 + - name: Test build_and_upload.sh + run : | + bash build_and_upload.sh - name: build jax stable image run : | project=tpu-prod-env-multipod @@ -43,8 +46,6 @@ jobs: bash docker_build_dependency_image.sh LOCAL_IMAGE_NAME=$local_image_name MODE=stable && docker tag ${local_image_name} gcr.io/$project/${cloud_image_name}:latest docker push gcr.io/$project/${cloud_image_name}:latest - - name: Test build_and_upload.sh - run : | - bash build_and_upload.sh + \ No newline at end of file