feat: Add preview CustomJob which can be run on persistent resource

PiperOrigin-RevId: 552895845
googleapis · Aug 1, 2023 · 56906b0 · 56906b0
1 parent b2edaf1
commit 56906b0
Show file tree

Hide file tree

Showing 3 changed files with 687 additions and 1 deletion.
diff --git a/google/cloud/aiplatform/jobs.py b/google/cloud/aiplatform/jobs.py
@@ -1154,7 +1154,7 @@ class DataLabelingJob(_Job):
     pass
 
 
-class CustomJob(_RunnableJob):
+class CustomJob(_RunnableJob, base.PreviewMixin):
     """Vertex AI Custom Job."""
 
     _resource_noun = "customJobs"
@@ -1165,6 +1165,7 @@ class CustomJob(_RunnableJob):
     _parse_resource_name_method = "parse_custom_job_path"
     _format_resource_name_method = "custom_job_path"
     _job_type = "training"
+    _preview_class = "google.cloud.aiplatform.aiplatform.preview.jobs.CustomJob"
 
     def __init__(
         self,