Trigger unit tests for docker images upload workflow

[xibinliu]

Description

• images will only be tagged to the current date when unit tests pass
• Images will only be tagged to "latest" when unit tests pass

Tests

See this manually triggered workflow run

Checklist

Before submitting this PR, please make sure (put X in square brackets):

• I have performed a self-review of my code. For an optional AI review, add the gemini-review label.
• I have necessary comments in my code, particularly in hard-to-understand areas.
• I have run end-to-end tests tests and provided workload links above if applicable.
• I have made or will make corresponding changes to the doc if needed, including adding new documentation pages to the relevant Table of Contents (toctree directive) as explained in our documentation.

[xibinliu]

@SurbhiJainUSC we need to discuss what tags do we want to add to the images before and after testing.

Current:

• before tests: ${{ github.run_id }}, and hashes of the repos
• after tests: "latest", and "<image_date>"

[codecov]

Codecov Report

✅ All modified and coverable lines are covered by tests.

📢 Thoughts on this report? Let us know!

[SurbhiJainUSC]

Why do we need a different workflow for testing the docker image?

Can we update build_and_push_docker_image.yml to something like this:

# https://github.com/AI-Hypercomputer/maxtext/blob/main/.github/workflows/build_and_push_docker_image.yml#L111
- name: Build and push Docker image

# Runs tests on docker image with tag as ${{ github.run_id }}
- name: Test Dokcer Image

# [Add tags such as ](https://github.com/AI-Hypercomputer/maxtext/blob/main/.github/workflows/build_and_push_docker_image.yml#L129)
- name: Add tags to Docker Image

[xibinliu]

Merged the jobs into build_and_push_docker_image.yml and now tests are executed immediately after the image has been pushed.

One caveat: since the tpu-post-training-nightly depends on tpu-post-training-stable, the whole workflow needs more time to finish because it now waits for the test to be done.

[SurbhiJainUSC]

Please rebase, this has been moved out of UploadDockerImages.yml

[xibinliu]

Done.

[SurbhiJainUSC]

Can we move testing the docker image job before this step?

[SurbhiJainUSC]

We should tag the image with appropriate tags once our testing passes.

[xibinliu]

Done