fix: install within job

notebooks/official/prediction/custom_batch_prediction_feature_filter.ipynb

@@ -154,10 +154,11 @@
       "outputs": [],
       "source": [
         "# Install the packages\n",
-        "! pip3 install --upgrade google-cloud-aiplatform \\\n",
-        "                                    google-cloud-storage \\\n",
-        "                                    google-cloud-bigquery \\\n",
-        "                                    pyarrow -q"
+        "! pip3 install --upgrade --quiet google-cloud-aiplatform \\\n",
+        "                                 google-cloud-storage \\\n",
+        "                                 google-cloud-bigquery \\\n",
+        "                                 pyarrow \\\n",
+        "                                 db-dtypes\n"
       ]
     },
     {
@@ -352,7 +353,7 @@
       },
       "outputs": [],
       "source": [
-        "BUCKET_URI = \"gs://your-bucket-name-unique\"  # @param {type:\"string\"}"
+        "BUCKET_URI = f\"gs://your-bucket-name-{PROJECT_ID}-unique\"  # @param {type:\"string\"}"
       ]
     },
     {
@@ -372,7 +373,7 @@
       },
       "outputs": [],
       "source": [
-        "! gsutil mb -l $REGION -p $PROJECT_ID $BUCKET_URI"
+        "! gsutil mb -l {REGION} -p {PROJECT_ID} {BUCKET_URI}"
       ]
     },
     {
@@ -1016,7 +1017,7 @@
         "    display_name=JOB_NAME,\n",
         "    script_path=\"task.py\",\n",
         "    container_uri=TRAIN_IMAGE,\n",
-        "    requirements=[\"google-cloud-bigquery>=2.20.0\", \"db-dtypes\"],\n",
+        "    requirements=[\"google-cloud-bigquery>=2.20.0\", \"db-dtypes\", \"protobuf==3.20.3\"],\n",
         "    model_serving_container_image_uri=DEPLOY_IMAGE,\n",
         ")\n",
         "\n",
@@ -1439,12 +1440,7 @@
         "\n",
         "To clean up all Google Cloud resources used in this project, you can [delete the Google Cloud 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 notebook:\n",
-        "\n",
-        "- Training Job\n",
-        "- Model\n",
-        "- Cloud Storage Bucket\n",
-        "- BigQuery Dataset"
+        "Otherwise, you can delete the individual resources you created in this notebook."
       ]
     },
     {
@@ -1461,6 +1457,9 @@
         "# Delete the training job\n",
         "job.delete()\n",
         "\n",
+        "# Delete the dataset\n",
+        "dataset.delete()\n",
+        "\n",
         "# Delete the model\n",
         "model.delete()\n",
         "\n",