From f11b598f9069f77e86631ada53941876aea010bc Mon Sep 17 00:00:00 2001 From: Andrew Ferlitsch Date: Tue, 23 Nov 2021 10:27:46 -0800 Subject: [PATCH] docs: Update name of BQ source parameter in samples (#859) * fix: issue 205934591 * fix:test * fix:test --- .../create_and_import_dataset_tabular_bigquery_sample.py | 4 ++-- .../create_and_import_dataset_tabular_bigquery_sample_test.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/samples/model-builder/create_and_import_dataset_tabular_bigquery_sample.py b/samples/model-builder/create_and_import_dataset_tabular_bigquery_sample.py index b7f4ea8013..38328ab652 100644 --- a/samples/model-builder/create_and_import_dataset_tabular_bigquery_sample.py +++ b/samples/model-builder/create_and_import_dataset_tabular_bigquery_sample.py @@ -18,13 +18,13 @@ # [START aiplatform_sdk_create_and_import_dataset_tabular_bigquery_sample] def create_and_import_dataset_tabular_bigquery_sample( - display_name: str, project: str, location: str, bigquery_source: str, + display_name: str, project: str, location: str, bq_source: str, ): aiplatform.init(project=project, location=location) dataset = aiplatform.TabularDataset.create( - display_name=display_name, bigquery_source=bigquery_source, + display_name=display_name, bq_source=bq_source, ) dataset.wait() diff --git a/samples/model-builder/create_and_import_dataset_tabular_bigquery_sample_test.py b/samples/model-builder/create_and_import_dataset_tabular_bigquery_sample_test.py index 6eefcf7702..6917eee9cf 100644 --- a/samples/model-builder/create_and_import_dataset_tabular_bigquery_sample_test.py +++ b/samples/model-builder/create_and_import_dataset_tabular_bigquery_sample_test.py @@ -24,7 +24,7 @@ def test_create_and_import_dataset_tabular_bigquery_sample( create_and_import_dataset_tabular_bigquery_sample.create_and_import_dataset_tabular_bigquery_sample( project=constants.PROJECT, location=constants.LOCATION, - bigquery_source=constants.BIGQUERY_SOURCE, + bq_source=constants.BIGQUERY_SOURCE, display_name=constants.DISPLAY_NAME, ) @@ -32,5 +32,5 @@ def test_create_and_import_dataset_tabular_bigquery_sample( project=constants.PROJECT, location=constants.LOCATION ) mock_create_tabular_dataset.assert_called_once_with( - display_name=constants.DISPLAY_NAME, bigquery_source=constants.BIGQUERY_SOURCE, + display_name=constants.DISPLAY_NAME, bq_source=constants.BIGQUERY_SOURCE, )