Skip to content

Commit 6114125

Browse files
committed
Rename BatchPredict Files
1 parent 23ca570 commit 6114125

File tree

3 files changed

+4
-4
lines changed

3 files changed

+4
-4
lines changed
File renamed without changes.
File renamed without changes.

genai/batch_prediction/test_batch_predict.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -20,8 +20,8 @@
2020

2121
import pytest
2222

23-
import submit_with_bq
24-
import submit_with_gcs
23+
import batchpredict_with_bq
24+
import batchpredict_with_gcs
2525

2626

2727
os.environ["GOOGLE_GENAI_USE_VERTEXAI"] = "True"
@@ -57,10 +57,10 @@ def gcs_output_uri():
5757

5858

5959
def 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

6464
def 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

0 commit comments

Comments
 (0)