feat: SDK support for model monitoring

google/cloud/aiplatform/__init__.py

@@ -51,6 +51,7 @@
     BatchPredictionJob,
     CustomJob,
     HyperparameterTuningJob,
+    ModelDeploymentMonitoringJob,
 )
 from google.cloud.aiplatform.pipeline_jobs import PipelineJob
 from google.cloud.aiplatform.tensorboard import (
@@ -70,6 +71,15 @@
     AutoMLTextTrainingJob,
     AutoMLVideoTrainingJob,
 )
+from google.cloud.aiplatform.model_monitoring import (
+    EmailAlertConfig,
+    EndpointSkewDetectionConfig,
+    EndpointDriftDetectionConfig,
+    EndpointExplanationConfig,
+    EndpointObjectiveConfig,
+    RandomSampleConfig,
+    ScheduleConfig,
+)
 from google.cloud.aiplatform import helpers
 
 """
@@ -123,7 +133,12 @@
     "CustomTrainingJob",
     "CustomContainerTrainingJob",
     "CustomPythonPackageTrainingJob",
+    "EmailAlertConfig",
     "Endpoint",
+    "EndpointDriftDetectionConfig",
+    "EndpointExplanationConfig",
+    "EndpointObjectiveConfig",
+    "EndpointSkewDetectionConfig",
     "EntityType",
     "Execution",
     "Experiment",
@@ -136,9 +151,12 @@
     "HyperparameterTuningJob",
     "Model",
     "ModelEvaluation",
+    "ModelDeploymentMonitoringJob",
     "PipelineJob",
     "PrivateEndpoint",
+    "RandomSampleConfig",
     "SequenceToSequencePlusForecastingTrainingJob",
+    "ScheduleConfig",
     "TabularDataset",
     "Tensorboard",
     "TensorboardExperiment",

google/cloud/aiplatform/compat/__init__.py

@@ -1,6 +1,6 @@
 # -*- coding: utf-8 -*-
 
-# Copyright 2021 Google LLC
+# Copyright 2022 Google LLC
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -94,6 +94,10 @@
     types.model = types.model_v1beta1
     types.model_evaluation = types.model_evaluation_v1beta1
     types.model_evaluation_slice = types.model_evaluation_slice_v1beta1
+    types.model_deployment_monitoring_job = (
+        types.model_deployment_monitoring_job_v1beta1
+    )
+    types.model_monitoring = types.model_monitoring_v1beta1
     types.model_service = types.model_service_v1beta1
     types.operation = types.operation_v1beta1
     types.pipeline_failure_policy = types.pipeline_failure_policy_v1beta1
@@ -179,6 +183,8 @@
     types.model = types.model_v1
     types.model_evaluation = types.model_evaluation_v1
     types.model_evaluation_slice = types.model_evaluation_slice_v1
+    types.model_deployment_monitoring_job = types.model_deployment_monitoring_job_v1
+    types.model_monitoring = types.model_monitoring_v1
     types.model_service = types.model_service_v1
     types.operation = types.operation_v1
     types.pipeline_failure_policy = types.pipeline_failure_policy_v1

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

@@ -61,7 +61,9 @@
     model as model_v1beta1,
     model_evaluation as model_evaluation_v1beta1,
     model_evaluation_slice as model_evaluation_slice_v1beta1,
+    model_deployment_monitoring_job as model_deployment_monitoring_job_v1beta1,
     model_service as model_service_v1beta1,
+    model_monitoring as model_monitoring_v1beta1,
     operation as operation_v1beta1,
     pipeline_failure_policy as pipeline_failure_policy_v1beta1,
     pipeline_job as pipeline_job_v1beta1,
@@ -125,7 +127,9 @@
     model as model_v1,
     model_evaluation as model_evaluation_v1,
     model_evaluation_slice as model_evaluation_slice_v1,
+    model_deployment_monitoring_job as model_deployment_monitoring_job_v1,
     model_service as model_service_v1,
+    model_monitoring as model_monitoring_v1,
     operation as operation_v1,
     pipeline_failure_policy as pipeline_failure_policy_v1,
     pipeline_job as pipeline_job_v1,
@@ -191,7 +195,9 @@
     model_v1,
     model_evaluation_v1,
     model_evaluation_slice_v1,
+    model_deployment_monitoring_job_v1,
     model_service_v1,
+    model_monitoring_v1,
     operation_v1,
     pipeline_failure_policy_v1beta1,
     pipeline_job_v1,
@@ -254,7 +260,9 @@
     model_v1beta1,
     model_evaluation_v1beta1,
     model_evaluation_slice_v1beta1,
+    model_deployment_monitoring_job_v1beta1,
     model_service_v1beta1,
+    model_monitoring_v1beta1,
     operation_v1beta1,
     pipeline_failure_policy_v1beta1,
     pipeline_job_v1beta1,