Build and Push Docker TPU Images #114

	name: Build and Push Docker TPU Images

	on:
	workflow_run:
	workflows: ["Run Tests"]
	types:
	- completed
	branches: [main]
	workflow_dispatch:

	jobs:
	build:
	runs-on: ubuntu-latest

	steps:
	- name: Checkout code
	uses: actions/checkout@v3

	- name: Set up Docker Buildx
	uses: docker/setup-buildx-action@v2

	- name: Cache Docker layers
	uses: actions/cache@v3
	with:
	path: /tmp/.buildx-cache
	key: ${{ runner.os }}-buildx-${{ github.sha }}
	restore-keys: \|
	${{ runner.os }}-buildx-

	- name: Get current date
	id: date
	run: echo "DATE=$(date +'%Y%m%d')" >> $GITHUB_ENV

	- name: Login to GitHub Container Registry
	uses: docker/login-action@v2
	with:
	registry: ghcr.io
	username: ${{ github.actor }}
	password: ${{ secrets.DOCKER_PUSH_TOKEN }}

	- name: Build and Push Docker image
	run: \|
	docker buildx build --file docker/tpu/Dockerfile.base --tag ghcr.io/${{ github.repository_owner }}/levanter-base:latest --tag ghcr.io/${{ github.repository_owner }}/levanter-base:${{ env.DATE }} --push .

	- name: Build and Push Incremental Docker image
	run: \|
	docker buildx build --file docker/tpu/Dockerfile.incremental --tag ghcr.io/${{ github.repository_owner }}/levanter-tpu:latest --tag ghcr.io/${{ github.repository_owner }}/levanter-tpu:${{ env.DATE }} --push .

Provide feedback