fix: Fixed BigQuery datasets that have colon in URI (#855)

* fix: Fixed BigQuery datasets that have colon in URI The colon-based "project:dataset.table" format is no longer supported. Fixes: #854 * Fixed the fix Co-authored-by: Gergely Imreh <imrehg@gmail.com> Co-authored-by: Karl Weinmeister <11586922+kweinmeister@users.noreply.github.com> Co-authored-by: sasha-gitg <44654632+sasha-gitg@users.noreply.github.com> Co-authored-by: Gergely Imreh <imrehg@gmail.com> Co-authored-by: Ivan Cheung <ivans.mailbox@gmail.com>
googleapis · Jan 11, 2022 · 153578f · 153578f
1 parent 4bfa11b
commit 153578f
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/google/cloud/aiplatform/datasets/column_names_dataset.py b/google/cloud/aiplatform/datasets/column_names_dataset.py
@@ -237,6 +237,12 @@ def _retrieve_bq_source_columns(
         if bq_table_uri.startswith(prefix):
             bq_table_uri = bq_table_uri[len(prefix) :]
 
+        # The colon-based "project:dataset.table" format is no longer supported:
+        # Invalid dataset ID "bigquery-public-data:chicago_taxi_trips".
+        # Dataset IDs must be alphanumeric (plus underscores and dashes) and must be at most 1024 characters long.
+        # Using dot-based "project.dataset.table" format instead.
+        bq_table_uri = bq_table_uri.replace(":", ".")
+
         client = bigquery.Client(project=project, credentials=credentials)
         table = client.get_table(bq_table_uri)
         schema = table.schema