@@ -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()

@@ -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,
 )