fix: project ID

notebooks/official/automl/automl_forecasting_bqml_arima_plus_comparison.ipynb

@@ -323,7 +323,7 @@
       },
       "outputs": [],
       "source": [
-        "! gsutil mb -l $REGION -p $PROJECT_ID $BUCKET_URI"
+        "! gsutil mb -l {REGION} -p {PROJECT_ID} {BUCKET_URI}"
       ]
     },
     {
@@ -356,6 +356,9 @@
       },
       "outputs": [],
       "source": [
+        "import sys\n",
+        "\n",
+        "IS_COLAB = \"google.colab\" in sys.modules\n",
         "if (\n",
         "    SERVICE_ACCOUNT == \"\"\n",
         "    or SERVICE_ACCOUNT is None\n",
@@ -484,15 +487,21 @@
         "\n",
         "# Must be same region as TRAINING_DATASET_BQ_PATH.\n",
         "client = bigquery.Client(project=PROJECT_ID)\n",
-        "bq_dataset = bigquery.Dataset(arima_dataset_path)\n",
-        "bq_dataset.location = DATA_REGION\n",
-        "bq_dataset = client.create_dataset(bq_dataset)\n",
+        "bq_dataset_pre = bigquery.Dataset(arima_dataset_path)\n",
+        "bq_dataset_pre.location = DATA_REGION\n",
+        "try:\n",
+        "    bq_dataset = client.create_dataset(bq_dataset_pre)\n",
+        "except:\n",
+        "    bq_dataset = client.get_dataset(bq_dataset_pre)\n",
         "print(f\"Created bigquery dataset {arima_dataset_path} in {DATA_REGION}\")\n",
         "\n",
         "# Make this the same region as the other dataset for easier comparisons.\n",
-        "bq_dataset = bigquery.Dataset(vertex_dataset_path)\n",
-        "bq_dataset.location = DATA_REGION\n",
-        "bq_dataset = client.create_dataset(bq_dataset)\n",
+        "bq_dataset_pre = bigquery.Dataset(vertex_dataset_path)\n",
+        "bq_dataset_pre.location = DATA_REGION\n",
+        "try:\n",
+        "    bq_dataset = client.create_dataset(bq_dataset_pre)\n",
+        "except:\n",
+        "    bq_dataset = client.get_dataset(bq_dataset_pre)\n",
         "print(f\"Created bigquery dataset {vertex_dataset_path} in {DATA_REGION}\")"
       ]
     },