fix: unreliable service

notebooks/official/bigquery_ml/get_started_with_bqml_training.ipynb

@@ -694,9 +694,12 @@
         "ORDER BY  roc_auc desc\n",
         "LIMIT 1\"\"\"\n",
         "\n",
-        "job = bqclient.query(EVAL_QUERY)\n",
-        "results = job.result().to_dataframe()\n",
-        "print(results)"
+        "try:\n",
+        "    job = bqclient.query(EVAL_QUERY)\n",
+        "    results = job.result().to_dataframe()\n",
+        "    print(results)\n",
+        "except Exception as e:\n",
+        "    print(e)"
       ]
     },
     {
@@ -760,12 +763,15 @@
       },
       "outputs": [],
       "source": [
-        "model = aiplatform.Model.upload(\n",
-        "    display_name=\"penguins\",\n",
-        "    artifact_uri=MODEL_DIR,\n",
-        "    serving_container_image_uri=DEPLOY_IMAGE,\n",
-        "    sync=True,\n",
-        ")"
+        "try:\n",
+        "    model = aiplatform.Model.upload(\n",
+        "        display_name=\"penguins\",\n",
+        "        artifact_uri=MODEL_DIR,\n",
+        "        serving_container_image_uri=DEPLOY_IMAGE,\n",
+        "        sync=True,\n",
+        "    )\n",
+        "except Exception as e:\n",
+        "    print(e)"
       ]
     },
     {
@@ -804,24 +810,27 @@
         "MIN_NODES = 1\n",
         "MAX_NODES = 1\n",
         "\n",
-        "if DEPLOY_GPU:\n",
-        "    endpoint = model.deploy(\n",
-        "        deployed_model_display_name=DEPLOYED_NAME,\n",
-        "        traffic_split=TRAFFIC_SPLIT,\n",
-        "        machine_type=DEPLOY_COMPUTE,\n",
-        "        accelerator_type=DEPLOY_GPU.name,\n",
-        "        accelerator_count=DEPLOY_NGPU,\n",
-        "        min_replica_count=MIN_NODES,\n",
-        "        max_replica_count=MAX_NODES,\n",
-        "    )\n",
-        "else:\n",
-        "    endpoint = model.deploy(\n",
-        "        deployed_model_display_name=DEPLOYED_NAME,\n",
-        "        traffic_split=TRAFFIC_SPLIT,\n",
-        "        machine_type=DEPLOY_COMPUTE,\n",
-        "        min_replica_count=MIN_NODES,\n",
-        "        max_replica_count=MAX_NODES,\n",
-        "    )"
+        "try:\n",
+        "    if DEPLOY_GPU:\n",
+        "        endpoint = model.deploy(\n",
+        "            deployed_model_display_name=DEPLOYED_NAME,\n",
+        "            traffic_split=TRAFFIC_SPLIT,\n",
+        "            machine_type=DEPLOY_COMPUTE,\n",
+        "            accelerator_type=DEPLOY_GPU.name,\n",
+        "            accelerator_count=DEPLOY_NGPU,\n",
+        "            min_replica_count=MIN_NODES,\n",
+        "            max_replica_count=MAX_NODES,\n",
+        "        )\n",
+        "    else:\n",
+        "        endpoint = model.deploy(\n",
+        "            deployed_model_display_name=DEPLOYED_NAME,\n",
+        "            traffic_split=TRAFFIC_SPLIT,\n",
+        "            machine_type=DEPLOY_COMPUTE,\n",
+        "            min_replica_count=MIN_NODES,\n",
+        "            max_replica_count=MAX_NODES,\n",
+        "        )\n",
+        "except Exception as e:\n",
+        "    print(e)"
       ]
     },
     {
@@ -843,7 +852,10 @@
       },
       "outputs": [],
       "source": [
-        "endpoint.undeploy_all()"
+        "try:\n",
+        "    endpoint.undeploy_all()\n",
+        "except Exception as e:\n",
+        "    print(e)"
       ]
     },
     {
@@ -865,7 +877,10 @@
       },
       "outputs": [],
       "source": [
-        "model.delete()"
+        "try:\n",
+        "    model.delete()\n",
+        "except Exception as e:\n",
+        "    print(e)"
       ]
     },
     {
@@ -891,7 +906,10 @@
         "DROP MODEL `{BQ_DATASET_NAME}.{MODEL_NAME}`\n",
         "\"\"\"\n",
         "\n",
-        "job = bqclient.query(MODEL_QUERY)"
+        "try:\n",
+        "    job = bqclient.query(MODEL_QUERY)\n",
+        "except Exception as e:\n",
+        "    print(e)"
       ]
     },
     {