Skip to content

Commit 1d5a060

Browse files
authored
Merge branch 'main' into release-please--branches--main
2 parents 4cfe487 + 39d4c2d commit 1d5a060

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

tests/unit/vertexai/test_batch_prediction.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -657,7 +657,7 @@ def test_submit_batch_prediction_job_with_invalid_input_dataset(self):
657657
invalid_bq_uris = ["bq://projectId.dataset1", "bq://projectId.dataset2"]
658658
with pytest.raises(
659659
ValueError,
660-
match=("Multiple Bigquery input datasets are not supported."),
660+
match=("Multiple BigQuery input datasets are not supported."),
661661
):
662662
batch_prediction.BatchPredictionJob.submit(
663663
source_model=_TEST_GEMINI_MODEL_NAME,

vertexai/batch_prediction/_batch_prediction.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -171,7 +171,7 @@ def submit(
171171
gcs_source = input_dataset
172172
elif first_input_uri.startswith("bq://"):
173173
if not isinstance(input_dataset, str):
174-
raise ValueError("Multiple Bigquery input datasets are not supported.")
174+
raise ValueError("Multiple BigQuery input datasets are not supported.")
175175
bigquery_source = input_dataset
176176
else:
177177
raise ValueError(

0 commit comments

Comments
 (0)