Merge branch 'main' into mlops_install

notebooks/community/ml_ops/stage1/mlops_data_management.ipynb

@@ -39,18 +39,15 @@
         "    </a>\n",
         "  </td>\n",
         "  <td>\n",
-        "        <a href=\"https://colab.research.google.com/github/GoogleCloudPlatform/vertex-ai-samples/blob/main/notebooks/community/ml_ops/stage1/mlops_data_management.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://console.cloud.google.com/vertex-ai/workbench/deploy-notebook?download_url=https://raw.githubusercontent.com/GoogleCloudPlatform/vertex-ai-samples/main/notebooks/community/ml_ops/stage1/mlops_data_management.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>\n",
-        "<br/><br/><br/>"
+        "<br/><br/><br/>\n",
+        "\n",
+        "*Note: This notebook is not supported for execution in Colab*"
       ]
     },
     {
@@ -169,21 +166,20 @@
         "if IS_WORKBENCH_NOTEBOOK:\n",
         "    USER_FLAG = \"--user\"\n",
         "\n",
-        "ONCE_ONLY = True\n",
+        "ONCE_ONLY = False\n",
         "if ONCE_ONLY:\n",
-        "    ! pip3 install -U tensorflow==2.5 $USER_FLAG -q\n",
-        "    ! pip3 install -U tensorflow-data-validation==1.2 $USER_FLAG -q\n",
-        "    ! pip3 install -U tensorflow-transform==1.2 $USER_FLAG -q\n",
-        "    ! pip3 install -U tensorflow-io==0.18 $USER_FLAG -q\n",
-        "    ! pip3 install --upgrade google-cloud-aiplatform[tensorboard] $USER_FLAG -q\n",
-        "    ! pip3 install --upgrade google-cloud-pipeline-components $USER_FLAG -q\n",
-        "    ! pip3 install --upgrade google-cloud-bigquery $USER_FLAG -q\n",
-        "    ! pip3 install --upgrade google-cloud-logging $USER_FLAG -q\n",
-        "    ! pip3 install --upgrade apache-beam[gcp]==2.33.0 $USER_FLAG -q\n",
-        "    ! pip3 install --upgrade pyarrow $USER_FLAG -q\n",
-        "    ! pip3 install --upgrade cloudml-hypertune $USER_FLAG -q\n",
-        "    ! pip3 install --upgrade kfp $USER_FLAG -q\n",
-        "    ! pip3 install future $USER_FLAG -q"
+        "    ! pip3 install -U {USER_FLAG} -q tensorflow==2.5 \\\n",
+        "                                     tensorflow-data-validation==1.2 \\\n",
+        "                                     tensorflow-transform==1.2 \\\n",
+        "                                     tensorflow-io==0.18 \n",
+        "    \n",
+        "    ! pip3 install --upgrade {USER_FLAG} -q google-cloud-aiplatform[tensorboard] \\\n",
+        "                                            google-cloud-pipeline-components \\\n",
+        "                                            google-cloud-bigquery \\\n",
+        "                                            google-cloud-logging \\\n",
+        "                                            apache-beam[gcp] \\\n",
+        "                                            pyarrow \\\n",
+        "                                            cloudml-hypertune\n"
       ]
     },
     {
@@ -355,7 +351,7 @@
       "source": [
         "### Authenticate your Google Cloud account\n",
         "\n",
-        "**If you are using Vertex AI Workbench Notebooks**, your environment is already authenticated. Skip this step.\n",
+        "**If you are using Vertex AI Workbench Notebooks**, your environment is already authenticated. \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",
@@ -417,7 +413,7 @@
         "\n",
         "**The following steps are required, regardless of your notebook environment.**\n",
         "\n",
-        "When you submit a custom training job using the Vertex SDK, you upload a Python package\n",
+        "When you submit a custom training job using the Vertex AI SDK, you upload a Python package\n",
         "containing your training code to a Cloud Storage bucket. Vertex AI runs\n",
         "the code from this package. In this tutorial, Vertex AI also saves the\n",
         "trained model that results from your job in the same bucket. You can then\n",