docs: Fix Create and Import Tabular BQ dataset sample

PiperOrigin-RevId: 502617257
googleapis · Jan 17, 2023 · 4415c10 · 4415c10
1 parent d06b22d
commit 4415c10
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
@@ -21,14 +21,14 @@ def create_and_import_dataset_tabular_bigquery_sample(
     display_name: str,
     project: str,
     location: str,
-    bigquery_source: str,
+    bq_source: str,
 ):
 
     aiplatform.init(project=project, location=location)
 
     dataset = aiplatform.TabularDataset.create(
         display_name=display_name,
-        bigquery_source=bigquery_source,
+        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,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,
     )
 
@@ -33,5 +33,5 @@ def test_create_and_import_dataset_tabular_bigquery_sample(
     )
     mock_create_tabular_dataset.assert_called_once_with(
         display_name=constants.DISPLAY_NAME,
-        bigquery_source=constants.BIGQUERY_SOURCE,
+        bq_source=constants.BIGQUERY_SOURCE,
     )