Made Model.batch_predict(job_display_name=) optional

googleapis · Feb 3, 2022 · fabb496 · fabb496
1 parent be0d97c
commit fabb496
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/google/cloud/aiplatform/jobs.py b/google/cloud/aiplatform/jobs.py
@@ -527,6 +527,8 @@ def create(
             (jobs.BatchPredictionJob):
                 Instantiated representation of the created batch prediction job.
         """
+        if not job_display_name:
+            job_display_name = cls._generate_display_name()
 
         utils.validate_display_name(job_display_name)
 

diff --git a/google/cloud/aiplatform/models.py b/google/cloud/aiplatform/models.py
@@ -2145,7 +2145,7 @@ def _deploy(
 
     def batch_predict(
         self,
-        job_display_name: str,
+        job_display_name: Optional[str] = None,
         gcs_source: Optional[Union[str, Sequence[str]]] = None,
         bigquery_source: Optional[str] = None,
         instances_format: str = "jsonl",
@@ -2182,7 +2182,7 @@ def batch_predict(
 
         Args:
             job_display_name (str):
-                Required. The user-defined name of the BatchPredictionJob.
+                Optional. The user-defined name of the BatchPredictionJob.
                 The name can be up to 128 characters long and can be consist
                 of any UTF-8 characters.
             gcs_source: Optional[Sequence[str]] = None