File tree Expand file tree Collapse file tree 3 files changed +4
-4
lines changed
Expand file tree Collapse file tree 3 files changed +4
-4
lines changed File renamed without changes.
File renamed without changes.
Original file line number Diff line number Diff line change 2020
2121import pytest
2222
23- import submit_with_bq
24- import submit_with_gcs
23+ import batchpredict_with_bq
24+ import batchpredict_with_gcs
2525
2626
2727os .environ ["GOOGLE_GENAI_USE_VERTEXAI" ] = "True"
@@ -57,10 +57,10 @@ def gcs_output_uri():
5757
5858
5959def test_batch_prediction_with_bq (bq_output_uri ) -> None :
60- response = submit_with_bq .generate_content (output_uri = bq_output_uri )
60+ response = batchpredict_with_bq .generate_content (output_uri = bq_output_uri )
6161 assert response == JobState .JOB_STATE_SUCCEEDED
6262
6363
6464def test_batch_prediction_with_gcs (gcs_output_uri ) -> None :
65- response = submit_with_gcs .generate_content (output_uri = gcs_output_uri )
65+ response = batchpredict_with_gcs .generate_content (output_uri = gcs_output_uri )
6666 assert response == JobState .JOB_STATE_SUCCEEDED
You can’t perform that action at this time.
0 commit comments