Fix: Upload docker images on pypi release#3487

copybara-service[bot] merged 1 commit intomainAI-Hypercomputer/maxtext:mainfrom

pypi_docker_fixAI-Hypercomputer/maxtext:pypi_docker_fixCopy head branch name to clipboard