xiaohei/model_garden
1
0
Fork 0
mirror of https://github.com/GoogleCloudPlatform/vertex-ai-samples.git synced 2026-09-26 14:42:04 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: CI/CD failure (#2123)

Browse Source 
* fix: CI/CD failure

* fix: delete dataset
This commit is contained in:
Andrew Ferlitsch
2023-07-28 23:17:40 +00:00
committed by GitHub
parent d48692bd4b
commit 3e89a23166
1 changed files with 14 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
notebooks/official/automl/sdk_automl_tabular_forecasting_batch.ipynb
+14 -3
View File
@@ -620,10 +620,15 @@
")\n",
"# Must be the same region as batch_predict_bq_input_uri\n",
"client = bigquery.Client(project=PROJECT_ID)\n",
"bq_dataset = bigquery.Dataset(batch_predict_bq_output_dataset_path)\n",
"bq_dataset_id = bigquery.Dataset(batch_predict_bq_output_dataset_path)\n",
"dataset_region = \"US\" # @param {type : \"string\"}\n",
"bq_dataset.location = dataset_region\n",
"bq_dataset = client.create_dataset(bq_dataset)\n",
"bq_dataset_id.location = dataset_region\n",
"# delete any existing dataset\n",
"try:\n",
" client.delete_dataset(bq_dataset_id, delete_contents=True)\n",
"except Exception as e:\n",
" print(e)\n",
"bq_dataset = client.create_dataset(bq_dataset_id)\n",
"print(\n",
" \"Created bigquery dataset {} in {}\".format(\n",
" batch_predict_bq_output_dataset_path, dataset_region\n",
@@ -857,6 +862,12 @@
"# Delete batch prediction job\n",
"batch_prediction_job.delete()\n",
"\n",
"# Delete the dataset\n",
"try:\n",
" client.delete_dataset(bq_dataset_id)\n",
"except Exception as e:\n",
" print(e)\n",
"\n",
"# Set this to true only if you'd like to delete your bucket\n",
"delete_bucket = False\n",
"\n",