Merge branch 'main' into model_monitoring_container

feat: add notebook using custom deployment container
2026-09-26 22:51:56 +00:00 · 2022-09-01 10:20:20 -07:00 · 2022-09-01 15:26:45 +00:00 · 2022-09-01 15:25:40 +00:00
43 changed files with 393 additions and 18463 deletions
@@ -1,6 +1,5 @@
 *  @vertex-ai-samples-contributors  @GoogleCloudPlatform/cloudml-samples-owners
 /tf_agents_bandits_movie_recommendation_with_kfp_and_vertex_sdk	@yinghsienwu
-/pytorch_pre_built_images_deployment @googleapis/vertex-prediction-team
 /pytorch_text_classification_using_vertex_sdk_and_gcloud  @RajeshThallam
 /pytorch_text_classification_using_vertex_sdk_and_gcloud		@RajeshThallam @ultrons
 /sklearn_text_classification_from_script_using_vertex_sdk       @maxhardt
@@ -2,5 +2,4 @@ cpr_model_server.py
 entrypoint.py
 state_dict.pth
 config.json
-**/__pycache__
-!testdata/**
+**/__pycache__
@@ -2,7 +2,7 @@

 ## About CPR

-CPR ([custom prediction routines](https://github.com/googleapis/python-aiplatform/blob/main/google/cloud/aiplatform/prediction/README.md)) is a framework designed by Google Cloud developers to make it easier to combine machine learning models with custom preprocessing and postprocessing logic in a real-time serving application. 
+CPR ([custom prediction routines](https://github.com/googleapis/python-aiplatform/blob/custom-prediction-routine/google/cloud/aiplatform/prediction/README.md)) is a framework designed by Google Cloud developers to make it easier to combine machine learning models with custom preprocessing and postprocessing logic in a real-time serving application. 

 ## Using this example

@@ -34,23 +34,6 @@ Finally, install the Python modules required to build and run the model server:
 pip install -r requirements.txt
 ```

-### Auth
-
-This example uses Google Cloud Storage for hosting model artifacts and Artifact Registry to store the container image. 
-You'll need to authorize yourself before you can interact with these.
-
-First, log in to GCP with application default credentials:
-```sh
-gcloud auth application-default login
-```
-
-Next, if you haven't done so already, set up the [gcloud credential helper](https://cloud.google.com/artifact-registry/docs/docker/authentication)
-for the Artifact Registry region where you intend to host the image.  
-```
-gcloud auth configure-docker <region>-docker.pkg.dev
-```
-
-
 ### Predictor

 The `TimmPredictor` class in `timm_serving/predictor.py` implements most of the important logic for the server.
@@ -60,9 +60,9 @@ class CPRConfig(object):
    image: str = "timm_predictor:latest"
    artifact_local_dir: str = ""
    region: str = "us-central1"
-    project_id: str = "<your project ID here>"
+    project_id: str = "samthrasher-experimental"
    repository: str = "cpr-images"
-    artifact_gcs_dir: str = "gs://<your bucket ID here>/timm-vit224/"
+    artifact_gcs_dir: str = "gs://samthrasher-cpr-example/timm-vit224/"
    model_name: str = ""
    endpoint_name: str = ""
    machine_type: str = "n1-standard-2"
@@ -5,4 +5,4 @@ timm==0.5.4
 smart_open==6.0.0

 google-cloud-storage>=1.26.0,<2.0.0dev
-google-cloud-aiplatform[prediction]>=1.16.0
+google-cloud-aiplatform[prediction] @ git+https://github.com/googleapis/python-aiplatform.git@custom-prediction-routine
@@ -70,10 +70,7 @@ class PredictorUnitTests(absltest.TestCase):
    def setUp(self):
        super().setUp()
        self.config = CPRConfig()
-        try:
-            self.config.load()
-        except FileNotFoundError:
-            logging.info("No saved config file found, using default values.")
+        self.config.load()
        self.predictor = predictor.TimmPredictor()

    def test_load_from_saved_state_dict_ok(self):
@@ -173,10 +170,7 @@ class ServerEndToEndTests(absltest.TestCase):
    def setUp(self):
        super().setUp()
        self.config = CPRConfig()
-        try:
-            self.config.load()
-        except FileNotFoundError:
-            logging.info("No saved config file found, using default values.")
+        self.config.load()
        self.local_model = cpr.LocalModel(
            serving_container_spec=aiplatform.gapic.ModelContainerSpec(
                image_uri=self.config.image
@@ -1 +0,0 @@
-blah
@@ -1,30 +0,0 @@
-# PyTorch Deployment on Google Cloud: Text Classification
-
-**This is an Experimental release**, covered by the Pre-GA Offerings Terms of your Google Cloud Platform [Terms of Service](https://cloud.google.com/terms).
-
-Experiments are focused on validating a prototype and are not guaranteed to be released. They are not intended for production use or covered by any SLA, support obligation, or deprecation policy and might be subject to backward-incompatible changes.
-
-**Kindly drop us a note before you run any scale tests.**
-
-**Do not hesitate to contact vertexai-prediction-preview-feedback@google.com if you have any questions or run into any issues.**
-
-The projects need to be added to the allowlist in order to deploy PyTorch models using Vertex AI Prediction pre-built PyTorch images. If you are interested in the feature, please send an email to vertexai-prediction-preview-feedback@google.com to provide your project numbers OR project ids.
-
-## Overview
-
-In the PyTorch on Google Cloud series of blog posts, we aim to share how to deploy PyTorch models at scale on [Vertex AI](https://cloud.google.com/vertex-ai).
-
-This tutorial on text classification shows how to deploy a PyTorch based text classification model on [Vertex AI](https://cloud.google.com/vertex-ai/docs/start/client-libraries#python) using Vertex SDK and [`gcloud ai`](https://cloud.google.com/sdk/gcloud/reference/beta/ai).
-
-## Notebooks
-
-| <h4>Notebook</h4> | <h4>Description</h4>                       |
-| :-------- | :------- |
-| [pytorch-text-classification-vertex-ai-deploy.ipynb](./pytorch-text-classification-vertex-ai-deploy.ipynb) | Notebook to show deploying a PyTorch model on Vertex AI |
-
-## Folders
-
-
-| <h4>Folder Name</h4> | <h4>Description</h4>                       |
-| :-------- | :------- |
-| [`predictor`](./predictor) | Folder with custom prediction handler to deploy a PyTorch model to Vertex Prediction. In the [notebook](./pytorch-text-classification-vertex-ai-deploy.ipynb), this folder is used for deploying a PyTorch model on Vertex AI using Vertex Prediction pre-built PyTorch images |
@@ -1,91 +0,0 @@
-
-import os
-import json
-import logging
-
-import torch
-from transformers import AutoModelForSequenceClassification, AutoTokenizer
-from ts.torch_handler.base_handler import BaseHandler
-
-logger = logging.getLogger(__name__)
-
-
-class TransformersClassifierHandler(BaseHandler):
-    """
-    The handler takes an input string and returns the classification text 
-    based on the serialized transformers checkpoint.
-    """
-    def __init__(self):
-        super(TransformersClassifierHandler, self).__init__()
-        self.initialized = False
-
-    def initialize(self, ctx):
-        """ Loads the model.pt file and initialized the model object.
-        Instantiates Tokenizer for preprocessor to use
-        Loads labels to name mapping file for post-processing inference response
-        """
-        self.manifest = ctx.manifest
-
-        properties = ctx.system_properties
-        model_dir = properties.get("model_dir")
-        self.device = torch.device("cuda:" + str(properties.get("gpu_id")) if torch.cuda.is_available() else "cpu")
-
-        # Read model serialize/pt file
-        serialized_file = self.manifest["model"]["serializedFile"]
-        model_pt_path = os.path.join(model_dir, serialized_file)
-        if not os.path.isfile(model_pt_path):
-            raise RuntimeError("Missing the model.pt or pytorch_model.bin file")
-        
-        # Load model
-        self.model = AutoModelForSequenceClassification.from_pretrained(model_dir)
-        self.model.to(self.device)
-        self.model.eval()
-        logger.debug('Transformer model from path {0} loaded successfully'.format(model_dir))
-        
-        # Ensure to use the same tokenizer used during training
-        self.tokenizer = AutoTokenizer.from_pretrained('bert-base-cased')
-
-        # Read the mapping file, index to object name
-        mapping_file_path = os.path.join(model_dir, "index_to_name.json")
-
-        if os.path.isfile(mapping_file_path):
-            with open(mapping_file_path) as f:
-                self.mapping = json.load(f)
-        else:
-            logger.warning('Missing the index_to_name.json file. Inference output will default.')
-            self.mapping = {"0": "Negative",  "1": "Positive"}
-
-        self.initialized = True
-
-    def preprocess(self, data):
-        """ Preprocessing input request by tokenizing
-            Extend with your own preprocessing steps as needed
-        """
-        text = data[0].get("data")
-        if text is None:
-            text = data[0].get("body")
-        sentences = text.decode('utf-8')
-        logger.info("Received text: '%s'", sentences)
-
-        # Tokenize the texts
-        tokenizer_args = ((sentences,))
-        inputs = self.tokenizer(*tokenizer_args,
-                                padding='max_length',
-                                max_length=128,
-                                truncation=True,
-                                return_tensors = "pt")
-        return inputs
-
-    def inference(self, inputs):
-        """ Predict the class of a text using a trained transformer model.
-        """
-        prediction = self.model(inputs['input_ids'].to(self.device))[0].argmax().item()
-
-        if self.mapping:
-            prediction = self.mapping[str(prediction)]
-
-        logger.info("Model predicted: '%s'", prediction)
-        return [prediction]
-
-    def postprocess(self, inference_output):
-        return inference_output
@@ -1,5 +0,0 @@
-
-{
-    "0": "Negative", 
-    "1": "Positive"
-}
@@ -17,7 +17,6 @@
 /explainable_ai/SDK_Custom_Container_XAI.ipynb @brianchunkang
 /matching_engine/sdk_matching_engine_for_indexing.ipynb @ivanmkc
 /matching_engine/matching_engine_for_indexing.ipynb @yinghsienwu
-/matching_engine/stream_update_for_matching_engine.ipynb @peterping666
 /sdk/pytorch_lightning_custom_container_training.ipynb @brianchunkang
 /tensorboard @yfang1
 /feature_store @nayaknishant @morgandu
@@ -32,18 +32,18 @@
        "<table align=\"left\">\n",
        "\n",
        "  <td>\n",
-        "    <a href=\"https://colab.research.google.com/github/GoogleCloudPlatform/vertex-ai-samples/blob/master/notebooks/official/bigquery_ml/bqml-online-prediction.ipynb\">\n",
+        "    <a href=\"https://colab.research.google.com/github/GoogleCloudPlatform/vertex-ai-samples/blob/master/notebooks/community/bigquery_ml/bqml-online-prediction.ipynb\">\n",
        "      <img src=\"https://cloud.google.com/ml-engine/images/colab-logo-32px.png\" alt=\"Colab logo\"> Run in Colab\n",
        "    </a>\n",
        "  </td>\n",
        "  <td>\n",
-        "    <a href=\"https://github/GoogleCloudPlatform/vertex-ai-samples/blob/master/notebooks/official/bigquery_ml/bqml-online-prediction.ipynb\">\n",
+        "    <a href=\"https://github/GoogleCloudPlatform/vertex-ai-samples/blob/master/notebooks/community/bigquery_ml/bqml-online-prediction.ipynb\">\n",
        "      <img src=\"https://cloud.google.com/ml-engine/images/github-logo-32px.png\" alt=\"GitHub logo\">\n",
        "      View on GitHub\n",
        "    </a>\n",
        "  </td>\n",
        "  <td>\n",
-        "    <a href=\"https://console.cloud.google.com/vertex-ai/workbench/deploy-notebook?download_url=https://raw.githubusercontent.com/GoogleCloudPlatform/vertex-ai-samples/blob/master/notebooks/official/bigquery_ml/bqml-online-prediction.ipynb\">\n",
+        "    <a href=\"https://console.cloud.google.com/vertex-ai/workbench/deploy-notebook?download_url=https://raw.githubusercontent.com/GoogleCloudPlatform/vertex-ai-samples/blob/master/notebooks/community/bigquery_ml/bqml-online-prediction.ipynb\">\n",
        "      <img src=\"https://lh3.googleusercontent.com/UiNooY4LUgW_oTvpsNhPpQzsstV5W8F7rYgxgGBD85cWJoLmrOzhVs_ksK_vgx40SHs7jCqkTkCk=e14-rj-sc0xffffff-h130-w32\" alt=\"Vertex AI logo\">\n",
        "      Open in Vertex AI Workbench\n",
        "    </a>\n",
@@ -63,7 +63,7 @@
        "\n",
        "### Dataset\n",
        "\n",
-        "The dataset, <a href=\"https://console.cloud.google.com/bigquery?project=bigquery-public-data&d=ga4_obfuscated_sample_ecommerce&p=bigquery-public-data&page=dataset\" target=\"_blank\">available publicly on BigQuery</a>, comes from obfuscated <a href=\"https://support.google.com/analytics/answer/10937659\" target=\"_blank\">Google Analytics 4 data</a> from the <a href=\"https://shop.googlemerchandisestore.com/\" target=\"_blank\">Google Merchandise Store</a>).\n",
+        "The dataset, [available publicly on BigQuery](https://console.cloud.google.com/bigquery?project=bigquery-public-data&d=ga4_obfuscated_sample_ecommerce&p=bigquery-public-data&page=dataset), comes from obfuscated [Google Analytics 4 data](https://support.google.com/analytics/answer/10937659) from the [Google Merchandise Store](https://shop.googlemerchandisestore.com/).\n",
        "\n",
        "### Objective\n",
        "\n",
@@ -95,9 +95,9 @@
        "* Vertex AI\n",
        "\n",
        "\n",
-        "Learn about <a href=\"https://cloud.google.com/bigquery/pricing\" target=\"_blank\">BigQuery Pricing</a>, <a href=\"https://cloud.google.com/bigquery-ml/pricing\" target=\"_blank\">BigQuery ML pricing</a>, <a href=\"https://cloud.google.com/vertex-ai/pricing\" target=\"_blank\">Vertex AI\n",
-        "pricing</a>, and use the <a href=\"https://cloud.google.com/products/calculator/\" target=\"_blank\">Pricing\n",
-        "Calculator</a>\n",
+        "Learn about [BigQuery Pricing](https://cloud.google.com/bigquery/pricing), [BigQuery ML pricing](https://cloud.google.com/bigquery-ml/pricing), [Vertex AI\n",
+        "pricing](https://cloud.google.com/vertex-ai/pricing), and use the [Pricing\n",
+        "Calculator](https://cloud.google.com/products/calculator/)\n",
        "to generate a cost estimate based on your projected usage."
      ]
    },
@@ -128,18 +128,18 @@
        "* virtualenv\n",
        "* Jupyter notebook running in a virtual environment with Python 3\n",
        "\n",
-        "The Google Cloud guide to <a href=\"https://cloud.google.com/python/setup\" target=\"_blank\">Setting up a Python development\n",
-        "environment</a> and the <a href=\"https://jupyter.org/install\" target=\"_blank\">Jupyter\n",
-        "installation guide</a> provide detailed instructions\n",
+        "The Google Cloud guide to [Setting up a Python development\n",
+        "environment](https://cloud.google.com/python/setup) and the [Jupyter\n",
+        "installation guide](https://jupyter.org/install) provide detailed instructions\n",
        "for meeting these requirements. The following steps provide a condensed set of\n",
        "instructions:\n",
        "\n",
-        "1. <a href=\"https://cloud.google.com/sdk/docs/\" target=\"_blank\">Install and initialize the Cloud SDK.</a>\n",
+        "1. [Install and initialize the Cloud SDK.](https://cloud.google.com/sdk/docs/)\n",
        "\n",
-        "1. <a href=\"https://cloud.google.com/python/setup#installing_python\" target=\"_blank\">Install Python 3.</a>\n",
+        "1. [Install Python 3.](https://cloud.google.com/python/setup#installing_python)\n",
        "\n",
-        "1. <a href=\"https://cloud.google.com/python/setup#installing_and_using_virtualenv\" target=\"_blank\">Install\n",
-        "   virtualenv</a>\n",
+        "1. [Install\n",
+        "   virtualenv](https://cloud.google.com/python/setup#installing_and_using_virtualenv)\n",
        "   and create a virtual environment that uses Python 3. Activate the virtual environment.\n",
        "\n",
        "1. To install Jupyter, run `pip3 install jupyter` on the\n",
@@ -234,13 +234,13 @@
        "\n",
        "**The following steps are required, regardless of your notebook environment.**\n",
        "\n",
-        "1. <a href=\"https://console.cloud.google.com/cloud-resource-manager\" target=\"_blank\">Select or create a Google Cloud project</a>. When you first create an account, you get a $300 free credit towards your compute/storage costs.\n",
+        "1. [Select or create a Google Cloud project](https://console.cloud.google.com/cloud-resource-manager). When you first create an account, you get a $300 free credit towards your compute/storage costs.\n",
        "\n",
-        "1. <a href=\"https://cloud.google.com/billing/docs/how-to/modify-project\" target=\"_blank\">Make sure that billing is enabled for your project</a>.\n",
+        "1. [Make sure that billing is enabled for your project](https://cloud.google.com/billing/docs/how-to/modify-project).\n",
        "\n",
-        "1. <a href=\"https://console.cloud.google.com/flows/enableapi?apiid=aiplatform.googleapis.com\" target=\"_blank\">Enable the Vertex AI API</a>.\n",
+        "1. [Enable the Vertex AI API](https://console.cloud.google.com/flows/enableapi?apiid=aiplatform.googleapis.com).\n",
        "\n",
-        "1. If you are running this notebook locally, you will need to install the <a href=\"https://cloud.google.com/sdk\" target=\"_blank\">Cloud SDK</a>.\n",
+        "1. If you are running this notebook locally, you will need to install the [Cloud SDK](https://cloud.google.com/sdk).\n",
        "\n",
        "1. Enter your project ID in the cell below. Then run the cell to make sure the\n",
        "Cloud SDK uses the right project for all the commands in this notebook.\n",
@@ -267,7 +267,7 @@
      },
      "outputs": [],
      "source": [
-        "PROJECT_ID = \"[YOUR-PROJECT-ID]\"\n",
+        "PROJECT_ID = \"YOUR-PROJECT-ID\"\n",
        "\n",
        "# Get your Google Cloud project ID from gcloud\n",
        "import os\n",
@@ -314,9 +314,9 @@
        "- Europe: `europe-west4`\n",
        "- Asia Pacific: `asia-east1`\n",
        "\n",
-        "You might not be able to use a multi-regional bucket for training with Vertex AI. Not all regions provide support for all Vertex AI services.\n",
+        "You may not use a multi-regional bucket for training with Vertex AI. Not all regions provide support for all Vertex AI services.\n",
        "\n",
-        "Learn more about <a href=\"https://cloud.google.com/vertex-ai/docs/general/locations\" target=\"_blank\">Vertex AI regions</a>."
+        "Learn more about [Vertex AI regions](https://cloud.google.com/vertex-ai/docs/general/locations)."
      ]
    },
    {
@@ -339,9 +339,9 @@
        "id": "06571eb4063b"
      },
      "source": [
-        "#### UUID\n",
+        "#### Timestamp\n",
        "\n",
-        "If you are in a live tutorial session, you might be using a shared test account or project. To avoid name collisions between users on resources created, you create a uuid for each instance session, and append it onto the name of resources you create in this tutorial."
+        "If you are in a live tutorial session, you might be using a shared test account or project. To avoid name collisions between users on resources created, you create a timestamp for each instance session, and append it onto the name of resources you create in this tutorial."
      ]
    },
    {
@@ -352,16 +352,9 @@
      },
      "outputs": [],
      "source": [
-        "import random\n",
-        "import string\n",
+        "from datetime import datetime\n",
        "\n",
-        "\n",
-        "# Generate a uuid of a specifed length(default=8)\n",
-        "def generate_uuid(length: int = 8) -> str:\n",
-        "    return \"\".join(random.choices(string.ascii_lowercase + string.digits, k=length))\n",
-        "\n",
-        "\n",
-        "UUID = generate_uuid()"
+        "TIMESTAMP = datetime.now().strftime(\"%Y%m%d%H%M%S\")"
      ]
    },
    {
@@ -387,7 +380,8 @@
        "\n",
        "**Otherwise**, follow these steps:\n",
        "\n",
-        "1. In the Cloud Console, go to the <a href=\"https://console.cloud.google.com/apis/credentials/serviceaccountkey\" target=\"_blank\">**Create service account key** page</a>.\n",
+        "1. In the Cloud Console, go to the [**Create service account key**\n",
+        "   page](https://console.cloud.google.com/apis/credentials/serviceaccountkey).\n",
        "\n",
        "2. Click **Create service account**.\n",
        "\n",
@@ -492,10 +486,7 @@
      },
      "outputs": [],
      "source": [
-        "from typing import Union\n",
-        "\n",
        "import google.cloud.aiplatform as vertex_ai\n",
-        "import pandas as pd\n",
        "from google.cloud import bigquery"
      ]
    },
@@ -559,17 +550,24 @@
      "outputs": [],
      "source": [
        "# Wrapper to use BigQuery client to run query/job, return job ID or result as DF\n",
-        "def run_bq_query(sql: str) -> Union[str, pd.DataFrame]:\n",
+        "def bq_query(sql):\n",
        "    \"\"\"\n",
        "    Input: SQL query, as a string, to execute in BigQuery\n",
        "    Returns the query results as a pandas DataFrame, or error, if any\n",
        "    \"\"\"\n",
+        "    # Import Exceptions library to help with dataset error catching\n",
+        "    from google.cloud.exceptions import BadRequest\n",
        "\n",
        "    # Try dry run before executing query to catch any errors\n",
-        "    job_config = bigquery.QueryJobConfig(dry_run=True, use_query_cache=False)\n",
-        "    bq_client.query(sql, job_config=job_config)\n",
+        "    try:\n",
+        "        job_config = bigquery.QueryJobConfig(dry_run=True, use_query_cache=False)\n",
+        "\n",
+        "        bq_client.query(sql, job_config=job_config)\n",
+        "\n",
+        "    except BadRequest as err:\n",
+        "        print(err)\n",
+        "        return\n",
        "\n",
-        "    # If dry run succeeds without errors, proceed to run query\n",
        "    job_config = bigquery.QueryJobConfig()\n",
        "    client_result = bq_client.query(sql, job_config=job_config)\n",
        "\n",
@@ -591,7 +589,7 @@
        "\n",
        "BigQuery ML (BQML) provides the capability to train ML tabular models, such as classification, regression, forecasting, and matrix factorization, in BigQuery using SQL syntax directly. BigQuery ML uses the scalable infrastructure of BigQuery ML so you don't need to set up additional infrastructure for training or batch serving.\n",
        "\n",
-        "Learn more about <a href=\"https://cloud.google.com/bigquery-ml/docs\" target=\"_blank\">BigQuery ML documentation</a>."
+        "Learn more about [BigQuery ML documentation](https://cloud.google.com/bigquery-ml/docs)."
      ]
    },
    {
@@ -602,13 +600,9 @@
      },
      "outputs": [],
      "source": [
-        "BQ_DATASET_NAME = f\"ga4_churnprediction_{UUID}\"\n",
+        "BQ_DATASET_NAME = \"ga4_churnprediction\"\n",
        "\n",
-        "sql_create_dataset = f\"\"\"CREATE SCHEMA IF NOT EXISTS {BQ_DATASET_NAME}\"\"\"\n",
-        "\n",
-        "print(sql_create_dataset)\n",
-        "\n",
-        "run_bq_query(sql_create_dataset)"
+        "bq_query(f\"\"\"CREATE SCHEMA IF NOT EXISTS {BQ_DATASET_NAME}\"\"\")"
      ]
    },
    {
@@ -626,7 +620,7 @@
        "id": "49dd00d5fbe5"
      },
      "source": [
-        "Inpect data that has been pre-processed from <a href=\"https://support.google.com/analytics/answer/10937659\" target=\"_blank\">Google Analytics 4 data from the Google Merchandise Store</a> so that it can be used for classification. For more information on how this data was prepared, read <a href=\"https://cloud.google.com/blog/topics/developers-practitioners/churn-prediction-game-developers-using-google-analytics-4-ga4-and-bigquery-ml\" target=\"_blank\">this blog post</a>.\n",
+        "Inpect data that has been pre-processed from [Google Analytics 4 data from the Google Merchandise Store](https://support.google.com/analytics/answer/10937659) so that it can be used for classification. For more information on how this data was prepared, read [this blog post](https://cloud.google.com/blog/topics/developers-practitioners/churn-prediction-game-developers-using-google-analytics-4-ga4-and-bigquery-ml).\n",
        "\n",
        "As seen below, each row represents a single user, and the columns represent their demographic features, their aggregated behavioral features in the first 24 hours of visiting the Google Merchandise Store, and the label (whether the user churned or returned any time after the first 24 hours)."
      ]
@@ -647,7 +641,7 @@
        "LIMIT\n",
        "    100\n",
        "\"\"\"\n",
-        "run_bq_query(sql_inspect)"
+        "bq_query(sql_inspect)"
      ]
    },
    {
@@ -668,9 +662,9 @@
        "The query below trains a logistic regression model using BigQuery ML. BigQuery resources are used to train the model.\n",
        "\n",
        "In the `OPTIONS` parameter:\n",
-        "* with `model_registry=\"vertex_ai\"`, the BigQuery ML model will automatically be <a href=\"https://cloud.google.com/vertex-ai/docs/model-registry/model-registry-bqml\" target=\"_blank\">registered to Vertex AI Model Registry</a>, which enables you to view all of your registered models and its versions on Google Cloud in one place.\n",
+        "* with `model_registry=\"vertex_ai\"`, the BigQuery ML model will automatically be [registered to Vertex AI Model Registry](https://cloud.google.com/vertex-ai/docs/model-registry/model-registry-bqml), which enables you to view all of your registered models and its versions on Google Cloud in one place.\n",
        "\n",
-        "* `vertex_ai_model_version_aliases allows you to set aliases to help you keep track of your model version (<a href=\"https://cloud.google.com/vertex-ai/docs/model-registry/model-alias\" target=\"_blank\">documentation</a>)."
+        "* `vertex_ai_model_version_aliases allows you to set aliases to help you keep track of your model version ([documentation](https://cloud.google.com/vertex-ai/docs/model-registry/model-alias))."
      ]
    },
    {
@@ -683,7 +677,7 @@
      "source": [
        "# this cell may take ~1 min to run\n",
        "\n",
-        "BQML_MODEL_NAME = f\"bqml_model_churn_{UUID}\"\n",
+        "BQML_MODEL_NAME = \"bqmlmodelchurn\"\n",
        "\n",
        "sql_train_model_bqml = f\"\"\"\n",
        "CREATE OR REPLACE MODEL {BQ_DATASET_NAME}.{BQML_MODEL_NAME}    \n",
@@ -702,7 +696,7 @@
        "\n",
        "print(sql_train_model_bqml)\n",
        "\n",
-        "run_bq_query(sql_train_model_bqml)"
+        "bq_query(sql_train_model_bqml)"
      ]
    },
    {
@@ -720,7 +714,7 @@
        "id": "2aaaae772f67"
      },
      "source": [
-        "With the model created, you can now evaluate the logistic regression model. Behind the scenes, BigQuery ML automatically <a href=\"https://cloud.google.com/bigquery-ml/docs/reference/standard-sql/bigqueryml-syntax-create#data_split_method\" target=\"_blank\">split the data</a>, which makes it easier to quickly train and evaluate models."
+        "With the model created, you can now evaluate the logistic regression model. Behind the scenes, BigQuery ML automatically [split the data](https://cloud.google.com/bigquery-ml/docs/reference/standard-sql/bigqueryml-syntax-create#data_split_method), which makes it easier to quickly train and evaluate models."
      ]
    },
    {
@@ -740,7 +734,7 @@
        "\n",
        "print(sql_evaluate_model)\n",
        "\n",
-        "run_bq_query(sql_evaluate_model)"
+        "bq_query(sql_evaluate_model)"
      ]
    },
    {
@@ -751,7 +745,7 @@
      "source": [
        "These metrics help you understand the performance of the model. \n",
        "\n",
-        "There are various metrics for logistic regression and other model types (full list of metrics can be found in the <a href=\"https://cloud.google.com/bigquery-ml/docs/reference/standard-sql/bigqueryml-syntax-evaluate#mlevaluate_output\" target=\"_blank\">documentation</a>)."
+        "There are various metrics for logistic regression and other model types (full list of metrics can be found in the [documentation](https://cloud.google.com/bigquery-ml/docs/reference/standard-sql/bigqueryml-syntax-evaluate#mlevaluate_output))."
      ]
    },
    {
@@ -771,7 +765,7 @@
      "source": [
        "Make a batch prediction in BigQuery ML on the original training data to check the probability of churn for each of the users, as seen in the `probability` column, with the predicted label under the `predicted_churn` column.\n",
        "\n",
-        "<a href=\"https://cloud.google.com/bigquery-ml/docs/reference/standard-sql/bigqueryml-syntax-explain-predict\" target=\"_blank\">ML.EXPLAIN_PREDICT</a> has built-in <a href=\"https://cloud.google.com/bigquery-ml/docs/reference/standard-sql/bigqueryml-syntax-xai-overview\" target=\"_blank\">Explainable AI</a>. This allows you to see the top contributing features to each prediction and interpret how it was computed."
+        "[ML.EXPLAIN_PREDICT](https://cloud.google.com/bigquery-ml/docs/reference/standard-sql/bigqueryml-syntax-explain-predict) has built-in [Explainable AI](https://cloud.google.com/bigquery-ml/docs/reference/standard-sql/bigqueryml-syntax-xai-overview). This allows you to see the top contributing features to each prediction and interpret how it was computed."
      ]
    },
    {
@@ -793,7 +787,7 @@
        "\n",
        "print(sql_explain_predict)\n",
        "\n",
-        "run_bq_query(sql_explain_predict)"
+        "bq_query(sql_explain_predict)"
      ]
    },
    {
@@ -802,7 +796,7 @@
        "id": "fa1f96c0f452"
      },
      "source": [
-        "Since the `top_feature_attributions` is a nested column, you can unnest the array (<a href=\"https://cloud.google.com/bigquery/docs/reference/standard-sql/arrays\" target=\"_blank\">documentation</a>) into separate rows for each of the features. In other words, since ML.EXPLAIN_PREDICT provides the top 5 most important features, using `UNNEST` results in 5 rows per prediction:"
+        "Since the `top_feature_attributions` is a nested column, you can unnest the array ([documentation](https://cloud.google.com/bigquery/docs/reference/standard-sql/arrays)) into separate rows for each of the features. In other words, since ML.EXPLAIN_PREDICT provides the top 5 most important features, using `UNNEST` results in 5 rows per prediction:"
      ]
    },
    {
@@ -833,7 +827,7 @@
        "\n",
        "print(sql_explain_predict)\n",
        "\n",
-        "run_bq_query(sql_explain_predict)"
+        "bq_query(sql_explain_predict)"
      ]
    },
    {
@@ -853,7 +847,7 @@
      "source": [
        "When the model was trained in BigQuery ML, the line `model_registry=\"vertex_ai\"` registered the model to Vertex AI Model Registry automatically upon completion.\n",
        "\n",
-        "You can view the model on the <a href=\"https://console.cloud.google.com/vertex-ai/models\" target=\"_blank\">Vertex AI Model Registry page</a>, or use the code below to check that it was successfully registered:"
+        "You can view the model on the [Vertex AI Model Registry page](https://console.cloud.google.com/vertex-ai/models), or use the code below to check that it was successfully registered:"
      ]
    },
    {
@@ -864,7 +858,12 @@
      },
      "outputs": [],
      "source": [
-        "model = vertex_ai.Model(model_name=BQML_MODEL_NAME)\n",
+        "print(f\"BQML_MODEL_NAME = {BQML_MODEL_NAME}\")\n",
+        "\n",
+        "models = vertex_ai.Model.list(\n",
+        "    filter=f\"display_name={BQML_MODEL_NAME}\", order_by=\"update_time\"\n",
+        ")\n",
+        "model = models[0]\n",
        "\n",
        "print(model.gca_resource)"
      ]
@@ -884,7 +883,7 @@
        "id": "b6120dcc1ff6"
      },
      "source": [
-        "While BigQuery ML supports batch prediction with <a href=\"https://cloud.google.com/bigquery-ml/docs/reference/standard-sql/bigqueryml-syntax-predict\" target=\"_blank\">ML.PREDICT</a> and <a href=\"https://cloud.google.com/bigquery-ml/docs/reference/standard-sql/bigqueryml-syntax-explain-predict\" target=\"_blank\">ML.EXPLAIN_PREDICT</a>, BigQuery ML is not suitable for real-time predictions where you need low latency predictions with potentially high frequency of requests.\n",
+        "While BigQuery ML supports batch prediction with [ML.PREDICT](https://cloud.google.com/bigquery-ml/docs/reference/standard-sql/bigqueryml-syntax-predict) and [ML.EXPLAIN_PREDICT](https://cloud.google.com/bigquery-ml/docs/reference/standard-sql/bigqueryml-syntax-explain-predict), BigQuery ML is not suitable for real-time predictions where you need low latency predictions with potentially high frequency of requests.\n",
        "\n",
        "In other words, deploying the BigQuery ML model to an endpoint enables you to do online predictions."
      ]
@@ -907,6 +906,30 @@
        "To deploy your model to an endpoint, you will first need to create an endpoint before you deploy the model to it."
      ]
    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "id": "3ce73125dff6"
+      },
+      "outputs": [],
+      "source": [
+        "def create_endpoint(\n",
+        "    project: str,\n",
+        "    display_name: str,\n",
+        "    location: str,\n",
+        "):\n",
+        "    endpoint = vertex_ai.Endpoint.create(\n",
+        "        display_name=display_name,\n",
+        "        project=project,\n",
+        "        location=location,\n",
+        "    )\n",
+        "\n",
+        "    print(endpoint.display_name)\n",
+        "    print(endpoint.resource_name)\n",
+        "    return endpoint"
+      ]
+    },
    {
      "cell_type": "code",
      "execution_count": null,
@@ -915,16 +938,17 @@
      },
      "outputs": [],
      "source": [
-        "ENDPOINT_NAME = f\"{BQML_MODEL_NAME}-endpoint\"\n",
+        "endpoint_name = f\"{BQML_MODEL_NAME}-{TIMESTAMP}\"\n",
        "\n",
-        "endpoint = vertex_ai.Endpoint.create(\n",
-        "    display_name=ENDPOINT_NAME,\n",
-        "    project=PROJECT_ID,\n",
-        "    location=REGION,\n",
+        "print(\n",
+        "    f\"\"\"\n",
+        "PROJECT_ID: {PROJECT_ID},\n",
+        "endpoint_name: {endpoint_name}\n",
+        "REGION: {REGION}\n",
+        "\"\"\"\n",
        ")\n",
        "\n",
-        "print(endpoint.display_name)\n",
-        "print(endpoint.resource_name)"
+        "create_endpoint(PROJECT_ID, endpoint_name, REGION)"
      ]
    },
    {
@@ -942,7 +966,31 @@
        "id": "951ed1693f6b"
      },
      "source": [
-        "List the endpoints to make sure it has successfully been created. (You can also view your endpoints on the <a href=\"https://console.cloud.google.com/vertex-ai/endpoints\" target=\"_blank\">Vertex AI Endpoints page</a>)."
+        "List the endpoints to make sure it has successfully been created. You can also view your endpoints on the [Vertex AI Endpoints page](https://console.cloud.google.com/vertex-ai/endpoints?project=polong-contentdev)."
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "id": "0a9bad8d9ad4"
+      },
+      "outputs": [],
+      "source": [
+        "endpoint = vertex_ai.Endpoint.list(\n",
+        "    # filter=f'display_name={endpoint_name}', # optional: filter by specific endpoint name\n",
+        "    order_by=\"update_time\"\n",
+        ")\n",
+        "endpoint[-1]"
+      ]
+    },
+    {
+      "cell_type": "markdown",
+      "metadata": {
+        "id": "2431a4d28d97"
+      },
+      "source": [
+        "Retrieve the endpoint id so you can use it in the next step."
      ]
    },
    {
@@ -953,7 +1001,7 @@
      },
      "outputs": [],
      "source": [
-        "endpoint.list()"
+        "endpoint[-1].to_dict()"
      ]
    },
    {
@@ -971,19 +1019,74 @@
        "id": "6a90be5b77a2"
      },
      "source": [
-        "With the new endpoint, you can now deploy your model."
+        "With the model, you can now deploy it to an endpoint. "
      ]
    },
    {
      "cell_type": "code",
      "execution_count": null,
      "metadata": {
-        "id": "c70ecc568ee5"
+        "id": "af323ea42c5b"
+      },
+      "outputs": [],
+      "source": [
+        "from typing import Dict, Optional, Sequence, Tuple\n",
+        "\n",
+        "\n",
+        "def deploy_model_with_automatic_resources_sample(\n",
+        "    project,\n",
+        "    location,\n",
+        "    model_name: str,\n",
+        "    endpoint: Optional[vertex_ai.Endpoint] = None,\n",
+        "    deployed_model_display_name: Optional[str] = None,\n",
+        "    traffic_percentage: Optional[int] = 0,\n",
+        "    traffic_split: Optional[Dict[str, int]] = None,\n",
+        "    min_replica_count: int = 1,\n",
+        "    max_replica_count: int = 1,\n",
+        "    metadata: Optional[Sequence[Tuple[str, str]]] = (),\n",
+        "    sync: bool = True,\n",
+        "):\n",
+        "    \"\"\"\n",
+        "    model_name: A fully-qualified model resource name or model ID.\n",
+        "          Example: \"projects/123/locations/us-central1/models/456\" or\n",
+        "          \"456\" when project and location are initialized or passed.\n",
+        "    \"\"\"\n",
+        "\n",
+        "    model = vertex_ai.Model(model_name=model_name)\n",
+        "\n",
+        "    model.deploy(\n",
+        "        endpoint=endpoint,\n",
+        "        deployed_model_display_name=deployed_model_display_name,\n",
+        "        traffic_percentage=traffic_percentage,\n",
+        "        traffic_split=traffic_split,\n",
+        "        min_replica_count=min_replica_count,\n",
+        "        max_replica_count=max_replica_count,\n",
+        "        metadata=metadata,\n",
+        "        sync=sync,\n",
+        "    )\n",
+        "\n",
+        "    model.wait()\n",
+        "\n",
+        "    print(model.display_name)\n",
+        "    print(model.resource_name)\n",
+        "    return"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "id": "9e6763369af4"
      },
      "outputs": [],
      "source": [
        "# deploying the model to the endpoint may take 10-15 minutes\n",
-        "model.deploy(endpoint=endpoint)"
+        "deploy_model_with_automatic_resources_sample(\n",
+        "    project=PROJECT_ID,\n",
+        "    location=REGION,\n",
+        "    model_name=BQML_MODEL_NAME,\n",
+        "    endpoint=endpoint[-1],\n",
+        ")"
      ]
    },
    {
@@ -992,7 +1095,7 @@
        "id": "c303d779477b"
      },
      "source": [
-        "You can also check on the status of your model by visiting the <a href=\"https://console.cloud.google.com/vertex-ai/endpoints\" target=\"_blank\">Vertex AI Endpoints page</a>."
+        "You can also check on the status of your model by visiting the [Vertex AI Endpoints page](https://console.cloud.google.com/vertex-ai/endpoints)."
      ]
    },
    {
@@ -1065,12 +1168,35 @@
      "cell_type": "code",
      "execution_count": null,
      "metadata": {
-        "id": "b4839f31d2f8"
+        "id": "2c6093ce9f8a"
      },
      "outputs": [],
      "source": [
-        "prediction = endpoint.predict(df_sample_requests_list)\n",
-        "print(prediction)"
+        "def endpoint_predict_sample(\n",
+        "    project: str, location: str, instances: list, endpoint: str\n",
+        "):\n",
+        "    endpoint = vertex_ai.Endpoint(endpoint)\n",
+        "\n",
+        "    prediction = endpoint.predict(instances=instances)\n",
+        "    return prediction"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "id": "0c41fd6eeb6f"
+      },
+      "outputs": [],
+      "source": [
+        "prediction_response = endpoint_predict_sample(\n",
+        "    project=PROJECT_ID,\n",
+        "    location=REGION,\n",
+        "    instances=df_sample_requests_list,\n",
+        "    endpoint=endpoint[-1].name,\n",
+        ")\n",
+        "\n",
+        "prediction_response"
      ]
    },
    {
@@ -1090,7 +1216,7 @@
      },
      "outputs": [],
      "source": [
-        "prediction.predictions"
+        "prediction_response.predictions"
      ]
    },
    {
@@ -1101,8 +1227,8 @@
      "source": [
        "## Cleaning up\n",
        "\n",
-        "To clean up all Google Cloud resources used in this project, you can <a href=\"https://cloud.google.com/resource-manager/docs/creating-managing-projects#shutting_down_projects\" target=\"_blank\">delete the Google Cloud\n",
-        "project</a> you used for the tutorial.\n",
+        "To clean up all Google Cloud resources used in this project, you can [delete the Google Cloud\n",
+        "project](https://cloud.google.com/resource-manager/docs/creating-managing-projects#shutting_down_projects) you used for the tutorial.\n",
        "\n",
        "Otherwise, you can delete the individual resources you created in this tutorial:"
      ]
@@ -1115,12 +1241,18 @@
      },
      "outputs": [],
      "source": [
-        "# Undeploy model from endpoint and delete endpoint\n",
-        "endpoint.undeploy_all()\n",
-        "endpoint.delete()\n",
+        "# MODEL_ID = model.name\n",
        "\n",
-        "# Delete BigQuery dataset, including the BigQuery ML model\n",
-        "! bq rm -r -f $PROJECT_ID:$BQ_DATASET_NAME"
+        "ENDPOINT_ID = int(endpoint[-1].name)\n",
+        "\n",
+        "# Undeploy model from endpoint\n",
+        "endpoint[-1].undeploy_all()\n",
+        "\n",
+        "# Delete endpoint resource\n",
+        "! gcloud ai endpoints delete $ENDPOINT_ID --quiet --region $REGION\n",
+        "\n",
+        "# Delete BigQuery ML model\n",
+        "! bq rm -f --model $PROJECT_ID\\:$BQ_DATASET_NAME\\.$BQML_MODEL_NAME"
      ]
    }
  ],
@@ -212,7 +212,7 @@
        "\n",
        "3. [Enable the Vertex AI APIs and Compute Engine APIs.](https://console.cloud.google.com/flows/enableapi?apiid=aiplatform.googleapis.com,compute_component)\n",
        "\n",
-        "4. [Google Cloud SDK](https://cloud.google.com/sdk) is already installed in Vertex AI Workbench Notebooks.\n",
+        "4. [Google Cloud SDK](https://cloud.google.com/sdk) is already installed in Google Cloud Notebooks.\n",
        "\n",
        "5. Enter your project ID in the cell below. Then run the  cell to make sure the\n",
        "Cloud SDK uses the right project for all the commands in this notebook.\n",
@@ -374,8 +374,15 @@
        "### Authenticate your Google Cloud account\n",
        "\n",
        "**If you are using Vertex AI Workbench Notebooks**, your environment is already\n",
-        "authenticated. \n",
-        "\n",
+        "authenticated. Skip this step."
+      ]
+    },
+    {
+      "cell_type": "markdown",
+      "metadata": {
+        "id": "32e1cd21a5d5"
+      },
+      "source": [
        "**If you are using Colab**, run the cell below and follow the instructions\n",
        "when prompted to authenticate your account via oAuth.\n",
        "\n",
@@ -741,16 +741,13 @@
        "\n",
        "### Input format for batch prediction jobs\n",
        "\n",
-        "The batch server accepts the following input formats for custom image models:\n",
+        "The batch server accepts the following input formats:\n",
        "\n",
        "- JSONL\n",
+        "- CSV\n",
+        "- TFRecords\n",
        "- File-List\n",
-        "\n",
-        "### Output format for batch prediction jobs\n",
-        "\n",
-        "The batch server accepts the following output formats for custom image models:\n",
-        "\n",
-        "- JSONL\n",
+        "- BigQuery table\n",
        "\n",
        "### Pivot format\n",
        "\n",
@@ -1265,8 +1262,16 @@
      "source": [
        "#### Prepare data for batch prediction\n",
        "\n",
+        "BLAH\n",
        "\n",
-        "Next, you format the same batch prediction request instances as a File-List format."
+        "Before you can run the data through batch prediction, you need to save the data into one of a few possible formats.\n",
+        "\n",
+        "For this tutorial, use JSONL as it's compatible with the 3-dimensional list that each image is currently represented in. To do this:\n",
+        "\n",
+        "1. In a file, write each instance as JSON on its own line.\n",
+        "2. Upload this file to Cloud Storage.\n",
+        "\n",
+        "For more details on batch prediction input formats: https://cloud.google.com/vertex-ai/docs/predictions/batch-predictions#batch_request_input"
      ]
    },
    {
@@ -1309,6 +1314,7 @@
      "source": [
        "### Send the prediction request\n",
        "\n",
+        "BLAH\n",
        "\n",
        "To make a batch prediction request, call the model object's `batch_predict` method with the following parameters: \n",
        "- `instances_format`: The format of the batch prediction request file: \"jsonl\", \"csv\", \"bigquery\", \"tf-record\", \"tf-record-gzip\" or \"file-list\"\n",
@@ -133,6 +133,8 @@
        "id": "8yVpQt-JHKPF"
      },
      "source": [
+        "## Before you begin\n",
+        "\n",
        "### Set up your local development environment\n",
        "\n",
        "**If you are using Colab or Vertex AI Workbench notebooks**, your environment already meets\n",
@@ -245,8 +247,6 @@
        "id": "BF1j6f9HApxa"
      },
      "source": [
-        "## Before you begin\n",
-        "\n",
        "### Set up your Google Cloud project\n",
        "\n",
        "**The following steps are required, regardless of your notebook environment.**\n",
@@ -1521,7 +1521,7 @@
  "metadata": {
    "colab": {
      "collapsed_sections": [],
-      "name": "get_started_with_model_monitoring_automl.ipynb",
+      "name": "model_monitoring_automl.ipynb",
      "toc_visible": true
    },
    "kernelspec": {
@@ -35,20 +35,20 @@
        "\n",
        "<table align=\"left\">\n",
        "  <td>\n",
-        "    <a href=\"https://colab.research.google.com/github/GoogleCloudPlatform/vertex-ai-samples/blob/main/notebooks/community/ml_ops/stage7/model_monitoring_custom.ipynb\">\n",
+        "    <a href=\"https://console.cloud.google.com/ai-platform/notebooks/deploy-notebook?name=Model%20Monitoring&download_url=https%3A%2F%2Fraw.githubusercontent.com%2FGoogleCloudPlatform%2Fvertex-ai-samples%2Fmain%2Fnotebooks%2Fofficial%2Fmodel_monitoring%2Fmodel_monitoring_custom.ipynb\">\n",
+        "       <img src=\"https://www.gstatic.com/cloud/images/navigation/vertex-ai.svg\" alt=\"Google Cloud Notebooks\">Open in Workbench AI Notebook\n",
+        "    </a>\n",
+        "  </td> \n",
+        "  <td>\n",
+        "    <a href=\"https://colab.research.google.com/github/GoogleCloudPlatform/vertex-ai-samples/blob/main/notebooks/official/model_monitoring/model_monitoring_custom.ipynb\">\n",
        "      <img src=\"https://cloud.google.com/ml-engine/images/colab-logo-32px.png\" alt=\"Colab logo\"> Open in Colab\n",
        "    </a>\n",
        "  </td>\n",
        "  <td>\n",
-        "    <a href=\"https://github.com/GoogleCloudPlatform/vertex-ai-samples/blob/main/notebooks/community/ml_ops/stage7/model_monitoring_custom.ipynb\">\n",
+        "    <a href=\"https://github.com/GoogleCloudPlatform/vertex-ai-samples/blob/main/notebooks/official/model_monitoring/model_monitoring_custom.ipynb\">\n",
        "        <img src=\"https://cloud.google.com/ml-engine/images/github-logo-32px.png\" alt=\"GitHub logo\">\n",
        "      View on GitHub\n",
        "    </a>\n",
-        "  </td> <td>\n",
-        "    <a href=\"https://console.cloud.google.com/vertex-ai/workbench/deploy-notebook?download_url=https://raw.githubusercontent.com/GoogleCloudPlatform/vertex-ai-samples/main/notebooks/community/ml_ops/stage7/model_monitoring_custom.ipynb\">\n",
-        "      <img src=\"https://lh3.googleusercontent.com/UiNooY4LUgW_oTvpsNhPpQzsstV5W8F7rYgxgGBD85cWJoLmrOzhVs_ksK_vgx40SHs7jCqkTkCk=e14-rj-sc0xffffff-h130-w32\" alt=\"Vertex AI logo\">\n",
-        "      Open in Vertex AI Workbench\n",
-        "    </a>\n",
        "  </td>\n",
        "</table>"
      ]
@@ -110,7 +110,7 @@
        "- identity - unique player identitity numbers\n",
        "- demographic features - information about the player, such as the geographic region in which a player is located\n",
        "- behavioral features - counts of the number of times a  player has triggered certain game events, such as reaching a new level\n",
-        "- churn propensity - this is the label or target feature, it provides an estimated probability that this player may churn, i.e. stop being an active player."
+        "- churn propensity - this is the label or target feature, it provides an estimated probability that this player will churn, i.e. stop being an active player."
      ]
    },
    {
@@ -140,6 +140,8 @@
        "id": "8yVpQt-JHKPF"
      },
      "source": [
+        "## Before you begin\n",
+        "\n",
        "### Set up your local development environment\n",
        "\n",
        "**If you are using Colab or Vertex AI Workbench notebooks**, your environment already meets\n",
@@ -252,8 +254,6 @@
        "id": "BF1j6f9HApxa"
      },
      "source": [
-        "## Before you begin\n",
-        "\n",
        "### Set up your Google Cloud project\n",
        "\n",
        "**The following steps are required, regardless of your notebook environment.**\n",
@@ -264,7 +264,7 @@
        "\n",
        "1. [Enable the Vertex AI API and Compute Engine API](https://console.cloud.google.com/flows/enableapi?apiid=aiplatform.googleapis.com,compute_component).\n",
        "\n",
-        "1. If you are running this notebook locally, you need to install the [Cloud SDK](https://cloud.google.com/sdk).\n",
+        "1. If you are running this notebook locally, you will need to install the [Cloud SDK](https://cloud.google.com/sdk).\n",
        "\n",
        "1. Enter your project ID in the cell below. Then run the cell to make sure the\n",
        "Cloud SDK uses the right project for all the commands in this notebook.\n",
@@ -780,7 +780,7 @@
        "\n",
        "When model monitoring is enabled, the sampled incoming prediction requests are logged into a BigQuery table. The input feature values contained in the logged requests are then analyzed for skew or drift on an specified interval basis. You set a sampling rate to monitor a subset of the production inputs to a model, and the monitoring interval.\n",
        "\n",
-        "The model monitoring service needs to know how to parse the feature values, which is referred to as the input schema. For AutoML tabular models, the input schema is automatically generated. For custom tabular models, the service attempts to automatically derive the input schema from the first 1000 prediction requests. Alternatively, one can upload the input schema.\n",
+        "The model monitoring service needs to know how to parse the feature values, which is referred to as the input schema. For AutoML tabular models, the input schema is automatically generated. For custom tabular models, the service will attempt to automatically derive the input schema from the first 1000 prediction requests. Alternatively, one can upload the input schema.\n",
        "\n",
        "For skew detection, the monitoring service requires a baseline for the statistical distribution of values in the training data. For AutoML tabular models this is automatically derived. For custom tabular models, you upload the training data to the service, and have the service automatically derive the distribution.\n",
        "\n",
@@ -1643,7 +1643,7 @@
  "metadata": {
    "colab": {
      "collapsed_sections": [],
-      "name": "get_started_with_model_monitoring_custom.ipynb",
+      "name": "model_monitoring_custom.ipynb",
      "toc_visible": true
    },
    "kernelspec": {
@@ -110,7 +110,7 @@
        "- identity - unique player identitity numbers\n",
        "- demographic features - information about the player, such as the geographic region in which a player is located\n",
        "- behavioral features - counts of the number of times a  player has triggered certain game events, such as reaching a new level\n",
-        "- churn propensity - this is the label or target feature, it provides an estimated probability that this player may churn, i.e. stop being an active player."
+        "- churn propensity - this is the label or target feature, it provides an estimated probability that this player will churn, i.e. stop being an active player."
      ]
    },
    {
@@ -140,6 +140,8 @@
        "id": "8yVpQt-JHKPF"
      },
      "source": [
+        "## Before you begin\n",
+        "\n",
        "### Set up your local development environment\n",
        "\n",
        "**If you are using Colab or Vertex AI Workbench notebooks**, your environment already meets\n",
@@ -252,8 +254,6 @@
        "id": "BF1j6f9HApxa"
      },
      "source": [
-        "## Before you begin\n",
-        "\n",
        "### Set up your Google Cloud project\n",
        "\n",
        "**The following steps are required, regardless of your notebook environment.**\n",
@@ -264,7 +264,7 @@
        "\n",
        "1. [Enable the Vertex AI API and Compute Engine API](https://console.cloud.google.com/flows/enableapi?apiid=aiplatform.googleapis.com,compute_component).\n",
        "\n",
-        "1. If you are running this notebook locally, you need to install the [Cloud SDK](https://cloud.google.com/sdk).\n",
+        "1. If you are running this notebook locally, you will need to install the [Cloud SDK](https://cloud.google.com/sdk).\n",
        "\n",
        "1. Enter your project ID in the cell below. Then run the cell to make sure the\n",
        "Cloud SDK uses the right project for all the commands in this notebook.\n",
@@ -944,7 +944,7 @@
        "\n",
        "When model monitoring is enabled, the sampled incoming prediction requests are logged into a BigQuery table. The input feature values contained in the logged requests are then analyzed for skew or drift on an specified interval basis. You set a sampling rate to monitor a subset of the production inputs to a model, and the monitoring interval.\n",
        "\n",
-        "The model monitoring service needs to know how to parse the feature values, which is referred to as the input schema. For AutoML tabular models, the input schema is automatically generated. For custom tabular models, the service attempts to automatically derive the input schema from the first 1000 prediction requests. Alternatively, one can upload the input schema.\n",
+        "The model monitoring service needs to know how to parse the feature values, which is referred to as the input schema. For AutoML tabular models, the input schema is automatically generated. For custom tabular models, the service will attempt to automatically derive the input schema from the first 1000 prediction requests. Alternatively, one can upload the input schema.\n",
        "\n",
        "For skew detection, the monitoring service requires a baseline for the statistical distribution of values in the training data. For AutoML tabular models this is automatically derived. For custom tabular models, you upload the training data to the service, and have the service automatically derive the distribution.\n",
        "\n",
@@ -1859,7 +1859,7 @@
  "metadata": {
    "colab": {
      "collapsed_sections": [],
-      "name": "get_started_with_model_monitoring_custom_tf_serving.ipynb",
+      "name": "model_monitoring_custom_tf_serving.ipynb",
      "toc_visible": true
    },
    "kernelspec": {
@@ -140,6 +140,8 @@
        "id": "8yVpQt-JHKPF"
      },
      "source": [
+        "## Before you begin\n",
+        "\n",
        "### Set up your local development environment\n",
        "\n",
        "**If you are using Colab or Vertex AI Workbench notebooks**, your environment already meets\n",
@@ -252,8 +254,6 @@
        "id": "BF1j6f9HApxa"
      },
      "source": [
-        "## Before you begin\n",
-        "\n",
        "### Set up your Google Cloud project\n",
        "\n",
        "**The following steps are required, regardless of your notebook environment.**\n",
@@ -1697,7 +1697,7 @@
  "metadata": {
    "colab": {
      "collapsed_sections": [],
-      "name": "get_started_with_model_monitoring_setup.ipynb",
+      "name": "model_monitoring_setup.ipynb",
      "toc_visible": true
    },
    "kernelspec": {
@@ -56,7 +56,7 @@ def parse_notebook(path):
        # cell 1 is copyright
        nth = 0
        cell, nth = get_cell(path, cells, nth)
-        if not 'Copyright' in cell['source'][0]:
+        if not cell['source'][0].startswith('# Copyright'):
            report_error(path, 0, "missing copyright cell")
            
        # check for notices
@@ -16,7 +16,6 @@
 /model_monitoring @andrewferlitsch
 /tensorboard @zbl94

-/bigquery_ml/bqml-online-prediction.ipynb @polong-lin
 /model_monitoring/model_monitoring.ipynb @mco-gh
 /ml_metadata/sdk-metric-parameter-tracking-for-custom-jobs.ipynb @jialuzh
 /ml_metadata/sdk-metric-parameter-tracking-for-locally-trained-models.ipynb @jialuzh
@@ -29,9 +28,6 @@
 /automl/automl_forecasting_bqml_arima_plus_comparison.ipynb @TheMichaelHu
 /automl/automl_tabular_on_vertex_pipelines.ipynb @helinwang
 /custom/custom_training_tensorboard_profiler.ipynb @itseric
-/workbench/spark/spark_sample_notebook.ipynb @bradmiro
-/workbench/spark/spark_ml.ipynb @bradmiro
+/workbench/spark/spark_sample_notebook.ipynb @bmiro
 /model-registry/bqml-vertexai-model-registry.ipynb @soheilazangeneh
 /workbench/exploratory_data_analysis/explore_data_in_bigquery_with_workbench.ipynb @alokpattani
-/model_evaluation/automl_tabular_classification_model_evaluation.ipynb @soheilazangeneh
-/model_evaluation/automl_tabular_regression_model_evaluation.ipynb @soheilazangeneh
@@ -8,7 +8,7 @@
      },
      "outputs": [],
      "source": [
-        "# Copyright 2021 Google LLC\n",
+        "# Copyright 2022 Google LLC\n",
        "#\n",
        "# Licensed under the Apache License, Version 2.0 (the \"License\");\n",
        "# you may not use this file except in compliance with the License.\n",
@@ -66,6 +66,17 @@
        "This tutorial demonstrates how to use the Vertex AI SDK to create image object detection models and do batch prediction using a Google Cloud [AutoML](https://cloud.google.com/vertex-ai/docs/start/automl-users) model."
      ]
    },
+    {
+      "cell_type": "markdown",
+      "metadata": {
+        "id": "dataset:salads,iod"
+      },
+      "source": [
+        "### Dataset\n",
+        "\n",
+        "The dataset used for this tutorial is the Salads category of the [OpenImages dataset](https://www.tensorflow.org/datasets/catalog/open_images_v4) from [TensorFlow Datasets](https://www.tensorflow.org/datasets/catalog/overview). This dataset does not require any feature engineering. The version of the dataset you will use in this tutorial is stored in a public Cloud Storage bucket. The trained model predicts the bounding box locations and the corresponding type of salad items in an image from a class of five items: salad, seafood, tomato, baked goods, or cheese."
+      ]
+    },
    {
      "cell_type": "markdown",
      "metadata": {
@@ -90,17 +101,6 @@
        "* Batch Prediction Service: Does a queued (batch) prediction for the entire set of instances in the background and stores the results in a Cloud Storage bucket when ready."
      ]
    },
-    {
-      "cell_type": "markdown",
-      "metadata": {
-        "id": "dataset:salads,iod"
-      },
-      "source": [
-        "### Dataset\n",
-        "\n",
-        "The dataset used for this tutorial is the Salads category of the [OpenImages dataset](https://www.tensorflow.org/datasets/catalog/open_images_v4) from [TensorFlow Datasets](https://www.tensorflow.org/datasets/catalog/overview). This dataset does not require any feature engineering. The version of the dataset you will use in this tutorial is stored in a public Cloud Storage bucket. The trained model predicts the bounding box locations and the corresponding type of salad items in an image from a class of five items: salad, seafood, tomato, baked goods, or cheese."
-      ]
-    },
    {
      "cell_type": "markdown",
      "metadata": {
@@ -201,7 +201,7 @@
      },
      "outputs": [],
      "source": [
-        "! pip3 install -U --upgrade tensorflow google-cloud-storage $USER_FLAG"
+        "! pip3 install -U google-cloud-storage $USER_FLAG"
      ]
    },
    {
@@ -213,6 +213,17 @@
        "Install the latest version of *tensorflow* library."
      ]
    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "id": "install_tensorflow"
+      },
+      "outputs": [],
+      "source": [
+        "! pip3 install --upgrade tensorflow $USER_FLAG"
+      ]
+    },
    {
      "cell_type": "markdown",
      "metadata": {
@@ -372,9 +383,9 @@
        "id": "timestamp"
      },
      "source": [
-        "#### UUID\n",
+        "#### Timestamp\n",
        "\n",
-        "If you are in a live tutorial session, you might be using a shared test account or project. To avoid name collisions between users on resources created, you create a uuid for each instance session, and append it onto the name of resources you create in this tutorial."
+        "If you are in a live tutorial session, you might be using a shared test account or project. To avoid name collisions between users on resources created, you create a timestamp for each instance session, and append the timestamp onto the name of resources you create in this tutorial."
      ]
    },
    {
@@ -385,16 +396,9 @@
      },
      "outputs": [],
      "source": [
-        "import random\n",
-        "import string\n",
+        "from datetime import datetime\n",
        "\n",
-        "\n",
-        "# Generate a uuid of a specifed length(default=8)\n",
-        "def generate_uuid(length: int = 8) -> str:\n",
-        "    return \"\".join(random.choices(string.ascii_lowercase + string.digits, k=length))\n",
-        "\n",
-        "\n",
-        "UUID = generate_uuid()"
+        "TIMESTAMP = datetime.now().strftime(\"%Y%m%d%H%M%S\")"
      ]
    },
    {
@@ -405,7 +409,7 @@
      "source": [
        "### Authenticate your Google Cloud account\n",
        "\n",
-        "**If you are using Google Cloud Notebooks**, your environment is already authenticated.\n",
+        "**If you are using Google Cloud Notebooks**, your environment is already authenticated. Skip this step.\n",
        "\n",
        "**If you are using Colab**, run the cell below and follow the instructions when prompted to authenticate your account via oAuth.\n",
        "\n",
@@ -490,7 +494,7 @@
      "outputs": [],
      "source": [
        "if BUCKET_NAME == \"\" or BUCKET_NAME is None or BUCKET_NAME == \"[your-bucket-name]\":\n",
-        "    BUCKET_URI = \"gs://\" + PROJECT_ID + \"aip-\" + UUID"
+        "    BUCKET_URI = \"gs://\" + PROJECT_ID + \"aip-\" + TIMESTAMP"
      ]
    },
    {
@@ -665,7 +669,7 @@
      "outputs": [],
      "source": [
        "dataset = aiplatform.ImageDataset.create(\n",
-        "    display_name=\"Salads\" + \"_\" + UUID,\n",
+        "    display_name=\"Salads\" + \"_\" + TIMESTAMP,\n",
        "    gcs_source=[IMPORT_FILE],\n",
        "    import_schema_uri=aiplatform.schema.dataset.ioformat.image.bounding_box,\n",
        ")\n",
@@ -713,7 +717,7 @@
      "outputs": [],
      "source": [
        "job = aiplatform.AutoMLImageTrainingJob(\n",
-        "    display_name=\"salads_\" + UUID,\n",
+        "    display_name=\"salads_\" + TIMESTAMP,\n",
        "    prediction_type=\"object_detection\",\n",
        "    multi_label=False,\n",
        "    model_type=\"CLOUD\",\n",
@@ -756,7 +760,7 @@
      "source": [
        "model = job.run(\n",
        "    dataset=dataset,\n",
-        "    model_display_name=\"salads_\" + UUID,\n",
+        "    model_display_name=\"salads_\" + TIMESTAMP,\n",
        "    training_fraction_split=0.8,\n",
        "    validation_fraction_split=0.1,\n",
        "    test_fraction_split=0.1,\n",
@@ -786,7 +790,7 @@
      "outputs": [],
      "source": [
        "# Get model resource ID\n",
-        "models = aiplatform.Model.list(filter=\"display_name=salads_\" + UUID)\n",
+        "models = aiplatform.Model.list(filter=\"display_name=salads_\" + TIMESTAMP)\n",
        "\n",
        "# Get a reference to the Model Service client\n",
        "client_options = {\"api_endpoint\": f\"{REGION}-aiplatform.googleapis.com\"}\n",
@@ -957,7 +961,7 @@
      "outputs": [],
      "source": [
        "batch_predict_job = model.batch_predict(\n",
-        "    job_display_name=\"salads_\" + UUID,\n",
+        "    job_display_name=\"salads_\" + TIMESTAMP,\n",
        "    gcs_source=gcs_input_uri,\n",
        "    gcs_destination_prefix=BUCKET_URI,\n",
        "    machine_type=\"n1-standard-4\",\n",
@@ -44,7 +44,7 @@
        "    </a>\n",
        "  </td>\n",
        "  <td>\n",
-        "    <a href=\"https://console.cloud.google.com/vertex-ai/workbench/deploy-notebook?download_url=https://raw.githubusercontent.com/GoogleCloudPlatform/vertex-ai-samples/main/notebooks/official/custom/custom-tabular-bq-managed-dataset.ipynb\">\n",
+        "    <a href=\"https://console.cloud.google.com/vertex-ai/workbench/deploy-notebook?download_url=https://github.com/GoogleCloudPlatform/vertex-ai-samples/blob/main/notebooks/official/custom/custom-tabular-bq-managed-dataset.ipynb\">\n",
        "      <img src=\"https://lh3.googleusercontent.com/UiNooY4LUgW_oTvpsNhPpQzsstV5W8F7rYgxgGBD85cWJoLmrOzhVs_ksK_vgx40SHs7jCqkTkCk=e14-rj-sc0xffffff-h130-w32\" alt=\"Vertex AI logo\">\n",
        "      Open in Vertex AI Workbench\n",
        "    </a>\n",
@@ -263,7 +263,7 @@
        "\n",
        "2. [Make sure that billing is enabled for your project](https://cloud.google.com/billing/docs/how-to/modify-project).\n",
        "\n",
-        "3. [Enable the following APIs: Vertex AI API, Cloud Resource Manager API](https://console.cloud.google.com/flows/enableapi?apiid=aiplatform.googleapis.com,cloudresourcemanager.googleapis.com).\n",
+        "3. [Enable the Vertex AI API](https://console.cloud.google.com/flows/enableapi?apiid=aiplatform.googleapis.com).\n",
        "\n",
        "4. If you are running this notebook locally, you need to install the [Cloud SDK](https://cloud.google.com/sdk).\n",
        "\n",
@@ -32,18 +32,18 @@
        "<table align=\"left\">\n",
        "\n",
        "  <td>\n",
-        "    <a href=\"https://colab.research.google.com/github/GoogleCloudPlatform/vertex-ai-samples/blob/main/notebooks/official/model-registry/bqml-vertexai-model-registry.ipynb\" target=\"_blank\">\n",
+        "    <a href=\"https://colab.research.google.com/github/GoogleCloudPlatform/vertex-ai-samples/blob/main/notebooks/official/bigquery-ml/bqml-vertexai-model-registry.ipynb\" target=\"_blank\">\n",
        "      <img src=\"https://cloud.google.com/ml-engine/images/colab-logo-32px.png\" alt=\"Colab logo\"> Run in Colab\n",
        "    </a>\n",
        "  </td>\n",
        "  <td>\n",
-        "    <a href=\"https://github.com/GoogleCloudPlatform/vertex-ai-samples/blob/main/notebooks/official/model-registry/bqml-vertexai-model-registry.ipynb\" target=\"_blank\">\n",
+        "    <a href=\"https://github.com/GoogleCloudPlatform/vertex-ai-samples/blob/main/notebooks/official/bigquery-ml/bqml-vertexai-model-registry.ipynb\" target=\"_blank\">\n",
        "      <img src=\"https://cloud.google.com/ml-engine/images/github-logo-32px.png\" alt=\"GitHub logo\">\n",
        "      View on GitHub\n",
        "    </a>\n",
        "  </td>\n",
        "  <td>\n",
-        "    <a href=\"https://console.cloud.google.com/vertex-ai/workbench/deploy-notebook?download_url=https://raw.githubusercontent.com/GoogleCloudPlatform/vertex-ai-samples/blob/main/notebooks/official/model-registry/bqml-vertexai-model-registry.ipynb\" target=\"_blank\">\n",
+        "    <a href=\"https://console.cloud.google.com/vertex-ai/workbench/deploy-notebook?download_url=https://raw.githubusercontent.com/GoogleCloudPlatform/vertex-ai-samples/blob/main/notebooks/official/bigquery-ml/bqml-vertexai-model-registry.ipynb\" target=\"_blank\">\n",
        "      <img src=\"https://lh3.googleusercontent.com/UiNooY4LUgW_oTvpsNhPpQzsstV5W8F7rYgxgGBD85cWJoLmrOzhVs_ksK_vgx40SHs7jCqkTkCk=e14-rj-sc0xffffff-h130-w32\" alt=\"Vertex AI logo\">\n",
        "      Open in Vertex AI Workbench\n",
        "    </a>\n",
@@ -677,7 +677,7 @@
      "source": [
        "### Find the model in the Vertex Model Registry\n",
        "\n",
-        "You can use the `Vertex AI Model()` method with `model_name` parameter to find the automatically registered model."
+        "You can use the `Vertex AI Model list()` method with a filter query to find the automatically registered model."
      ]
    },
    {
@@ -223,22 +223,12 @@
        "if IS_WORKBENCH_NOTEBOOK:\n",
        "    USER_FLAG = \"--user\"\n",
        "\n",
-        "# Don't bother installing tensorflow or explainable_ai_sdk on Colab\n",
-        "extra_pkgs = \"tensorflow explainable_ai_sdk\"\n",
-        "if \"google.colab\" in sys.modules:\n",
-        "  extra_pkgs = \"\"\n",
-        "\n",
        "# Install required packages.\n",
-        "! pip3 install {USER_FLAG} \\\n",
-        "    google-cloud-aiplatform \\\n",
-        "    explainable_ai_sdk \\\n",
-        "    $extra_pkgs \\\n",
-        "    google-api-python-client \\\n",
-        "    google-auth-oauthlib \\\n",
-        "    google-auth-httplib2 \\\n",
-        "    oauth2client \\\n",
-        "    requests \\\n",
-        "    google-cloud-storage==1.32.0"
+        "! pip3 install {USER_FLAG} --quiet --upgrade google-cloud-aiplatform\n",
+        "! pip3 install {USER_FLAG} --quiet --upgrade tensorflow\n",
+        "! pip3 install {USER_FLAG} --quiet --upgrade explainable_ai_sdk\n",
+        "! pip3 install {USER_FLAG} --quiet --upgrade google-api-python-client google-auth-oauthlib google-auth-httplib2 oauth2client requests\n",
+        "! pip3 install {USER_FLAG} --quiet --upgrade google-cloud-storage==1.32.0"
      ]
    },
    {
@@ -572,7 +562,7 @@
      "cell_type": "code",
      "execution_count": null,
      "metadata": {
-        "id": "wGa5T9eRR8Mz"
+        "id": "init_aip:mbsdk,region"
      },
      "outputs": [],
      "source": [
@@ -81,6 +81,7 @@
        "The steps performed include:\n",
        "\n",
        "- Define and compile a `Vertex AI` pipeline.\n",
+        "- Schedule a recurring pipeline run.\n",
        "- Specify which service account to use for a pipeline run."
      ]
    },
@@ -96,9 +97,13 @@
        "\n",
        "* Vertex AI\n",
        "* Cloud Storage\n",
+        "* Cloud Functions\n",
+        "* Cloud Scheduler\n",
        "\n",
        "Learn about [Vertex AI pricing](https://cloud.google.com/vertex-ai/pricing),\n",
        "[Cloud Storage pricing](https://cloud.google.com/storage/pricing),\n",
+        "[Cloud Functions pricing](ttps://cloud.google.com/functions/pricing), and\n",
+        "[Clould Scheduler pricing]((https://cloud.google.com/scheduler/pricing)),\n",
        "and use the [Pricing\n",
        "Calculator](https://cloud.google.com/products/calculator/)\n",
        "to generate a cost estimate based on your projected usage."
@@ -921,6 +926,69 @@
        "job.delete()"
      ]
    },
+    {
+      "cell_type": "markdown",
+      "metadata": {
+        "id": "schedule_pipeline_run"
+      },
+      "source": [
+        "## Recurring pipeline runs: create a scheduled pipeline job\n",
+        "\n",
+        "This section shows how to create a **scheduled pipeline job**.  You do this using the pipeline you already defined.\n",
+        "\n",
+        "Under the hood, the scheduled jobs are supported by the Cloud Scheduler and a Cloud Functions function.  Check first that the APIs for both of these services are enabled.\n",
+        "You will need to first enable the [enable the Cloud Scheduler API](http://console.cloud.google.com/apis/library/cloudscheduler.googleapis.com) and the [Cloud Functions and Cloud Build APIs](https://console.cloud.google.com/flows/enableapi?apiid=cloudfunctions,cloudbuild.googleapis.com) if you have not already done so.\n",
+        "Note:you need to [create an App Engine app for your project](https://cloud.google.com/scheduler/docs/quickstart) if one does not already exist.\n",
+        "\n",
+        "\n",
+        "See the [Cloud Scheduler](https://cloud.google.com/scheduler/docs/configuring/cron-job-schedules) documentation for more on the cron syntax.\n",
+        "\n",
+        "Create a scheduled pipeline job, passing as an argument the job specification file that you compiled above.\n",
+        "\n",
+        "*Note:* You can pass a `parameter_values` dict that specifies the pipeline input parameters you want to use."
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "id": "Ty5hDoNX2Ou8"
+      },
+      "outputs": [],
+      "source": [
+        "if not os.getenv(\"IS_TESTING\"):\n",
+        "    from kfp.v2.google.client import AIPlatformClient  # noqa: F811\n",
+        "\n",
+        "    api_client = AIPlatformClient(project_id=PROJECT_ID, region=REGION)\n",
+        "\n",
+        "    # adjust time zone and cron schedule as necessary\n",
+        "    response = api_client.create_schedule_from_job_spec(\n",
+        "        job_spec_path=\"intro_pipeline.json\",\n",
+        "        schedule=\"2 * * * *\",\n",
+        "        time_zone=\"America/Los_Angeles\",  # change this as necessary\n",
+        "        parameter_values={\"text\": \"Hello world!\"},\n",
+        "        # pipeline_root=PIPELINE_ROOT  # this argument is necessary if you did not specify PIPELINE_ROOT as part of the pipeline definition.\n",
+        "    )"
+      ]
+    },
+    {
+      "cell_type": "markdown",
+      "metadata": {
+        "id": "J8AP1viy2Ou8"
+      },
+      "source": [
+        "Once the scheduled job is created, you can see it listed in the [Cloud Scheduler](https://console.cloud.google.com/cloudscheduler/) panel in the Console.\n",
+        "\n",
+        "<a href=\"https://storage.googleapis.com/amy-jo/images/kf-pls/pipelines_scheduler.png\" target=\"_blank\"><img src=\"https://storage.googleapis.com/amy-jo/images/kf-pls/pipelines_scheduler.png\" width=\"95%\"/></a>\n",
+        "\n",
+        "You can test the setup from the Cloud Scheduler panel by clicking 'RUN NOW'.\n",
+        "\n",
+        "> **Note**: The implementation is using a Cloud Functions function, which you can see listed in the [Cloud Functions](https://console.cloud.google.com/functions/list) panel in the console as `templated_http_request-v1`.\n",
+        "Don't delete this function, as it will prevent the Cloud Scheduler jobs from actually kicking off the pipeline run.  If you do delete it, create a new scheduled job in order to recreate the function.\n",
+        "\n",
+        "When you're done experimenting, you probably want to **PAUSE** your scheduled job from the Cloud Scheduler panel, so that the recurrent jobs do not keep running."
+      ]
+    },
    {
      "cell_type": "markdown",
      "metadata": {
@@ -1154,7 +1222,7 @@
      "outputs": [],
      "source": [
        "delete_pipeline = True\n",
-        "delete_bucket = False\n",
+        "delete_bucket = True\n",
        "\n",
        "try:\n",
        "    if delete_pipeline and \"DISPLAY_NAME\" in globals():\n",
Author	SHA1	Message	Date
Andrew FerlitschandGitHub	5a1285d1a3	Merge branch 'main' into model_monitoring_container	2022-09-01 10:20:20 -07:00
Andrew Ferlitsch	26fad6b647	feat: add notebook using custom deployment container	2022-09-01 15:26:45 +00:00
Andrew Ferlitsch	779e695dd6	feat: add notebook using custom deployment container	2022-09-01 15:25:40 +00:00