Skip to content

Commit 1d1b3f0

Browse files
vertex-sdk-botcopybara-github
authored andcommitted
feat: Allow users to specify the job_display_name.
PiperOrigin-RevId: 716586588
1 parent 9e6cc62 commit 1d1b3f0

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

vertexai/batch_prediction/_batch_prediction.py

+6
Original file line numberDiff line numberDiff line change
@@ -115,6 +115,7 @@ def submit(
115115
input_dataset: Union[str, List[str]],
116116
*,
117117
output_uri_prefix: Optional[str] = None,
118+
job_display_name: Optional[str] = None,
118119
) -> "BatchPredictionJob":
119120
"""Submits a batch prediction job for a GenAI model.
120121
@@ -137,6 +138,10 @@ def submit(
137138
be used for GCS source and
138139
f"bq://projectId.gen_ai_batch_prediction.predictions_{TIMESTAMP}"
139140
will be used for BigQuery source.
141+
job_display_name (str):
142+
The user-defined name of the BatchPredictionJob.
143+
The name can be up to 128 characters long and can be consist
144+
of any UTF-8 characters.
140145
141146
Returns:
142147
Instantiated BatchPredictionJob.
@@ -209,6 +214,7 @@ def submit(
209214
try:
210215
aiplatform_job = jobs.BatchPredictionJob.submit(
211216
model_name=model_name,
217+
job_display_name=job_display_name,
212218
gcs_source=gcs_source,
213219
bigquery_source=bigquery_source,
214220
gcs_destination_prefix=gcs_destination_prefix,

0 commit comments

Comments
 (0)