docs: Update name of BQ source parameter in samples (#859)

* fix: issue 205934591 * fix:test * fix:test
googleapis · Nov 23, 2021 · f11b598 · f11b598
1 parent b613b26
commit f11b598
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 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
@@ -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
@@ -24,13 +24,13 @@ 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,
     )
 
     mock_sdk_init.assert_called_once_with(
         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,
     )