Skip to content

Commit 56906b0

Browse files
yinghsienwucopybara-github
authored andcommitted
feat: Add preview CustomJob which can be run on persistent resource
PiperOrigin-RevId: 552895845
1 parent b2edaf1 commit 56906b0

File tree

3 files changed

+687
-1
lines changed

3 files changed

+687
-1
lines changed

google/cloud/aiplatform/jobs.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -1154,7 +1154,7 @@ class DataLabelingJob(_Job):
11541154
pass
11551155

11561156

1157-
class CustomJob(_RunnableJob):
1157+
class CustomJob(_RunnableJob, base.PreviewMixin):
11581158
"""Vertex AI Custom Job."""
11591159

11601160
_resource_noun = "customJobs"
@@ -1165,6 +1165,7 @@ class CustomJob(_RunnableJob):
11651165
_parse_resource_name_method = "parse_custom_job_path"
11661166
_format_resource_name_method = "custom_job_path"
11671167
_job_type = "training"
1168+
_preview_class = "google.cloud.aiplatform.aiplatform.preview.jobs.CustomJob"
11681169

11691170
def __init__(
11701171
self,

0 commit comments

Comments
 (0)