Skip to content

Commit 4415c10

Browse files
sasha-gitgcopybara-github
authored andcommitted
docs: Fix Create and Import Tabular BQ dataset sample
PiperOrigin-RevId: 502617257
1 parent d06b22d commit 4415c10

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

samples/model-builder/create_and_import_dataset_tabular_bigquery_sample.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -21,14 +21,14 @@ def create_and_import_dataset_tabular_bigquery_sample(
2121
display_name: str,
2222
project: str,
2323
location: str,
24-
bigquery_source: str,
24+
bq_source: str,
2525
):
2626

2727
aiplatform.init(project=project, location=location)
2828

2929
dataset = aiplatform.TabularDataset.create(
3030
display_name=display_name,
31-
bigquery_source=bigquery_source,
31+
bq_source=bq_source,
3232
)
3333

3434
dataset.wait()

samples/model-builder/create_and_import_dataset_tabular_bigquery_sample_test.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,7 @@ def test_create_and_import_dataset_tabular_bigquery_sample(
2424
create_and_import_dataset_tabular_bigquery_sample.create_and_import_dataset_tabular_bigquery_sample(
2525
project=constants.PROJECT,
2626
location=constants.LOCATION,
27-
bigquery_source=constants.BIGQUERY_SOURCE,
27+
bq_source=constants.BIGQUERY_SOURCE,
2828
display_name=constants.DISPLAY_NAME,
2929
)
3030

@@ -33,5 +33,5 @@ def test_create_and_import_dataset_tabular_bigquery_sample(
3333
)
3434
mock_create_tabular_dataset.assert_called_once_with(
3535
display_name=constants.DISPLAY_NAME,
36-
bigquery_source=constants.BIGQUERY_SOURCE,
36+
bq_source=constants.BIGQUERY_SOURCE,
3737
)

0 commit comments

Comments
 (0)