Skip to content

Commit 62b8b23

Browse files
vertex-sdk-botcopybara-github
authored andcommitted
feat: Scheduled pipelines client GA.
test: Fix scheduled pipeline client system test. chore: Change list_jobs() `enable_simple_view` to default to True. chore: Change member variable `cron_expression` to `cron`. PiperOrigin-RevId: 557652030
1 parent fa2953b commit 62b8b23

File tree

14 files changed

+1257
-534
lines changed

14 files changed

+1257
-534
lines changed

google/cloud/aiplatform/__init__.py

+4
Original file line numberDiff line numberDiff line change
@@ -56,6 +56,9 @@
5656
ModelDeploymentMonitoringJob,
5757
)
5858
from google.cloud.aiplatform.pipeline_jobs import PipelineJob
59+
from google.cloud.aiplatform.pipeline_job_schedules import (
60+
PipelineJobSchedule,
61+
)
5962
from google.cloud.aiplatform.tensorboard import (
6063
Tensorboard,
6164
TensorboardExperiment,
@@ -167,6 +170,7 @@
167170
"ModelEvaluation",
168171
"ModelDeploymentMonitoringJob",
169172
"PipelineJob",
173+
"PipelineJobSchedule",
170174
"PrivateEndpoint",
171175
"RandomSampleConfig",
172176
"SequenceToSequencePlusForecastingTrainingJob",

google/cloud/aiplatform/compat/__init__.py

+3
Original file line numberDiff line numberDiff line change
@@ -144,6 +144,7 @@
144144
services.model_service_client = services.model_service_client_v1
145145
services.pipeline_service_client = services.pipeline_service_client_v1
146146
services.prediction_service_client = services.prediction_service_client_v1
147+
services.schedule_service_client = services.schedule_service_client_v1
147148
services.specialist_pool_service_client = services.specialist_pool_service_client_v1
148149
services.tensorboard_service_client = services.tensorboard_service_client_v1
149150
services.index_service_client = services.index_service_client_v1
@@ -208,6 +209,8 @@
208209
types.pipeline_state = types.pipeline_state_v1
209210
types.prediction_service = types.prediction_service_v1
210211
types.publisher_model = types.publisher_model_v1
212+
types.schedule = types.schedule_v1
213+
types.schedule_service = types.schedule_service_v1
211214
types.specialist_pool = types.specialist_pool_v1
212215
types.specialist_pool_service = types.specialist_pool_service_v1
213216
types.study = types.study_v1

google/cloud/aiplatform/compat/services/__init__.py

+4
Original file line numberDiff line numberDiff line change
@@ -109,6 +109,9 @@
109109
from google.cloud.aiplatform_v1.services.prediction_service import (
110110
client as prediction_service_client_v1,
111111
)
112+
from google.cloud.aiplatform_v1.services.schedule_service import (
113+
client as schedule_service_client_v1,
114+
)
112115
from google.cloud.aiplatform_v1.services.specialist_pool_service import (
113116
client as specialist_pool_service_client_v1,
114117
)
@@ -133,6 +136,7 @@
133136
model_service_client_v1,
134137
pipeline_service_client_v1,
135138
prediction_service_client_v1,
139+
schedule_service_client_v1,
136140
specialist_pool_service_client_v1,
137141
tensorboard_service_client_v1,
138142
vizier_service_client_v1,

google/cloud/aiplatform/compat/types/__init__.py

+4
Original file line numberDiff line numberDiff line change
@@ -145,6 +145,8 @@
145145
pipeline_state as pipeline_state_v1,
146146
prediction_service as prediction_service_v1,
147147
publisher_model as publisher_model_v1,
148+
schedule as schedule_v1,
149+
schedule_service as schedule_service_v1,
148150
specialist_pool as specialist_pool_v1,
149151
specialist_pool_service as specialist_pool_service_v1,
150152
study as study_v1,
@@ -215,6 +217,8 @@
215217
pipeline_state_v1,
216218
prediction_service_v1,
217219
publisher_model_v1,
220+
schedule_v1,
221+
schedule_service_v1,
218222
specialist_pool_v1,
219223
specialist_pool_service_v1,
220224
tensorboard_v1,

google/cloud/aiplatform/preview/constants/schedules.py google/cloud/aiplatform/constants/schedule.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@
1616
#
1717

1818
from google.cloud.aiplatform.compat.types import (
19-
schedule_v1beta1 as gca_schedule,
19+
schedule as gca_schedule,
2020
)
2121
from google.cloud.aiplatform.constants import pipeline as pipeline_constants
2222

0 commit comments

Comments
 (0)