cloudbuild.yaml

# These are Cloud Build steps. You can read more about cloud build here: 
# https://cloud.google.com/cloud-build/docs/
# All these steps sharing a filesystem between each other.
timeout: 2050s
steps:
  # Executing the notebook
- name: 'us-west2-docker.pkg.dev/public-gcr/ai-workbench/workbench-cli'
  id: 'run-clean-notebook'
  args: ['execute-notebook', '--notebook','./notebooks/mnist_training_loop_0223.ipynb', '--gcs-folder', 'gs://gtc-conf-examples/${COMMIT_SHA}/', '--location', 'us-west1', '--project', 'ml-lab-152505', '--wait', '--env-container', 'gcr.io/deeplearning-platform-release/pytorch-gpu.1-10']
  # Create model from the notebook
- name: 'us-west2-docker.pkg.dev/public-gcr/ai-workbench/workbench-cli'
  id: 'extract-model-from-notebook'
  args: ['extract-model-from-notebook', '--tag','us-west2-docker.pkg.dev/public-gcr/ai-workbench/test-model:${COMMIT_SHA}', '--src', './notebooks/', '--main-notebook', './notebooks/mnist_training_loop_0223.ipynb', '--target', './notebooks/container/', '--generate-only']
  # Bulding containers
- name: 'gcr.io/cloud-builders/docker'
  args: [ 'build', '-t', 'us-west2-docker.pkg.dev/public-gcr/ai-workbench/test-model:${COMMIT_SHA}', '-f', './notebooks/container/Dockerfile', './notebooks/container/' ]
  # Bulding containers
- name: 'gcr.io/cloud-builders/docker'
  args: [ 'push', 'us-west2-docker.pkg.dev/public-gcr/ai-workbench/test-model:${COMMIT_SHA}']