Make Docker build and upload scripts runnable through pip#3359

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

bvandermoon-docker-buildsAI-Hypercomputer/maxtext:bvandermoon-docker-buildsCopy head branch name to clipboard