Skip to content

Commit

Permalink
feat: Scheduled pipelines client GA.
Browse files Browse the repository at this point in the history
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
  • Loading branch information
vertex-sdk-bot authored and copybara-github committed Aug 17, 2023
1 parent fa2953b commit 62b8b23
Show file tree
Hide file tree
Showing 14 changed files with 1,257 additions and 534 deletions.
4 changes: 4 additions & 0 deletions google/cloud/aiplatform/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -56,6 +56,9 @@
ModelDeploymentMonitoringJob,
)
from google.cloud.aiplatform.pipeline_jobs import PipelineJob
from google.cloud.aiplatform.pipeline_job_schedules import (
PipelineJobSchedule,
)
from google.cloud.aiplatform.tensorboard import (
Tensorboard,
TensorboardExperiment,
Expand Down Expand Up @@ -167,6 +170,7 @@
"ModelEvaluation",
"ModelDeploymentMonitoringJob",
"PipelineJob",
"PipelineJobSchedule",
"PrivateEndpoint",
"RandomSampleConfig",
"SequenceToSequencePlusForecastingTrainingJob",
Expand Down
3 changes: 3 additions & 0 deletions google/cloud/aiplatform/compat/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -144,6 +144,7 @@
services.model_service_client = services.model_service_client_v1
services.pipeline_service_client = services.pipeline_service_client_v1
services.prediction_service_client = services.prediction_service_client_v1
services.schedule_service_client = services.schedule_service_client_v1
services.specialist_pool_service_client = services.specialist_pool_service_client_v1
services.tensorboard_service_client = services.tensorboard_service_client_v1
services.index_service_client = services.index_service_client_v1
Expand Down Expand Up @@ -208,6 +209,8 @@
types.pipeline_state = types.pipeline_state_v1
types.prediction_service = types.prediction_service_v1
types.publisher_model = types.publisher_model_v1
types.schedule = types.schedule_v1
types.schedule_service = types.schedule_service_v1
types.specialist_pool = types.specialist_pool_v1
types.specialist_pool_service = types.specialist_pool_service_v1
types.study = types.study_v1
Expand Down
4 changes: 4 additions & 0 deletions google/cloud/aiplatform/compat/services/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -109,6 +109,9 @@
from google.cloud.aiplatform_v1.services.prediction_service import (
client as prediction_service_client_v1,
)
from google.cloud.aiplatform_v1.services.schedule_service import (
client as schedule_service_client_v1,
)
from google.cloud.aiplatform_v1.services.specialist_pool_service import (
client as specialist_pool_service_client_v1,
)
Expand All @@ -133,6 +136,7 @@
model_service_client_v1,
pipeline_service_client_v1,
prediction_service_client_v1,
schedule_service_client_v1,
specialist_pool_service_client_v1,
tensorboard_service_client_v1,
vizier_service_client_v1,
Expand Down
4 changes: 4 additions & 0 deletions google/cloud/aiplatform/compat/types/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -145,6 +145,8 @@
pipeline_state as pipeline_state_v1,
prediction_service as prediction_service_v1,
publisher_model as publisher_model_v1,
schedule as schedule_v1,
schedule_service as schedule_service_v1,
specialist_pool as specialist_pool_v1,
specialist_pool_service as specialist_pool_service_v1,
study as study_v1,
Expand Down Expand Up @@ -215,6 +217,8 @@
pipeline_state_v1,
prediction_service_v1,
publisher_model_v1,
schedule_v1,
schedule_service_v1,
specialist_pool_v1,
specialist_pool_service_v1,
tensorboard_v1,
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,7 @@
#

from google.cloud.aiplatform.compat.types import (
schedule_v1beta1 as gca_schedule,
schedule as gca_schedule,
)
from google.cloud.aiplatform.constants import pipeline as pipeline_constants

Expand Down
Loading

0 comments on commit 62b8b23

Please sign in to comment.