feat: add RaySepc to ResourceRuntimeSpec, and add ResourceRuntime to PersistentResource

docs/aiplatform_v1/schedule_service.rst

@@ -0,0 +1,10 @@
+ScheduleService
+---------------------------------
+
+.. automodule:: google.cloud.aiplatform_v1.services.schedule_service
+    :members:
+    :inherited-members:
+
+.. automodule:: google.cloud.aiplatform_v1.services.schedule_service.pagers
+    :members:
+    :inherited-members:

docs/aiplatform_v1/services.rst

@@ -17,6 +17,7 @@ Services for Google Cloud Aiplatform v1 API
     model_service
     pipeline_service
     prediction_service
+    schedule_service
     specialist_pool_service
     tensorboard_service
     vizier_service

google/cloud/aiplatform/v1/schema/trainingjob/definition_v1/types/automl_tables.py

@@ -110,6 +110,7 @@ class AutoMlTablesInputs(proto.Message):
             the prediction type. If the field is not set, a
             default objective function is used.
             classification (binary):
+
               "maximize-au-roc" (default) - Maximize the
             area under the receiver
             operating characteristic (ROC) curve.
@@ -122,9 +123,11 @@ class AutoMlTablesInputs(proto.Message):
             Maximize recall for a specified
             precision value.
             classification (multi-class):
+
               "minimize-log-loss" (default) - Minimize log
             loss.
             regression:
+
               "minimize-rmse" (default) - Minimize
             root-mean-squared error (RMSE).   "minimize-mae"
             - Minimize mean-absolute error (MAE).

google/cloud/aiplatform/v1beta1/schema/trainingjob/definition_v1beta1/types/automl_tables.py

@@ -110,6 +110,7 @@ class AutoMlTablesInputs(proto.Message):
             the prediction type. If the field is not set, a
             default objective function is used.
             classification (binary):
+
               "maximize-au-roc" (default) - Maximize the
             area under the receiver
             operating characteristic (ROC) curve.
@@ -122,9 +123,11 @@ class AutoMlTablesInputs(proto.Message):
             Maximize recall for a specified
             precision value.
             classification (multi-class):
+
               "minimize-log-loss" (default) - Minimize log
             loss.
             regression:
+
               "minimize-rmse" (default) - Minimize
             root-mean-squared error (RMSE).   "minimize-mae"
             - Minimize mean-absolute error (MAE).

google/cloud/aiplatform_v1/__init__.py

@@ -50,6 +50,8 @@
 from .services.pipeline_service import PipelineServiceAsyncClient
 from .services.prediction_service import PredictionServiceClient
 from .services.prediction_service import PredictionServiceAsyncClient
+from .services.schedule_service import ScheduleServiceClient
+from .services.schedule_service import ScheduleServiceAsyncClient
 from .services.specialist_pool_service import SpecialistPoolServiceClient
 from .services.specialist_pool_service import SpecialistPoolServiceAsyncClient
 from .services.tensorboard_service import TensorboardServiceClient
@@ -463,8 +465,19 @@
 from .types.prediction_service import PredictRequest
 from .types.prediction_service import PredictResponse
 from .types.prediction_service import RawPredictRequest
+from .types.prediction_service import StreamingPredictRequest
+from .types.prediction_service import StreamingPredictResponse
 from .types.publisher_model import PublisherModel
 from .types.saved_query import SavedQuery
+from .types.schedule import Schedule
+from .types.schedule_service import CreateScheduleRequest
+from .types.schedule_service import DeleteScheduleRequest
+from .types.schedule_service import GetScheduleRequest
+from .types.schedule_service import ListSchedulesRequest
+from .types.schedule_service import ListSchedulesResponse
+from .types.schedule_service import PauseScheduleRequest
+from .types.schedule_service import ResumeScheduleRequest
+from .types.schedule_service import UpdateScheduleRequest
 from .types.service_networking import PrivateServiceConnectConfig
 from .types.specialist_pool import SpecialistPool
 from .types.specialist_pool_service import CreateSpecialistPoolOperationMetadata
@@ -544,6 +557,7 @@
 from .types.types import DoubleArray
 from .types.types import Int64Array
 from .types.types import StringArray
+from .types.types import Tensor
 from .types.unmanaged_container_model import UnmanagedContainerModel
 from .types.user_action_reference import UserActionReference
 from .types.value import Value
@@ -585,6 +599,7 @@
     "ModelServiceAsyncClient",
     "PipelineServiceAsyncClient",
     "PredictionServiceAsyncClient",
+    "ScheduleServiceAsyncClient",
     "SpecialistPoolServiceAsyncClient",
     "TensorboardServiceAsyncClient",
     "VizierServiceAsyncClient",
@@ -674,6 +689,7 @@
     "CreateModelDeploymentMonitoringJobRequest",
     "CreateNasJobRequest",
     "CreatePipelineJobRequest",
+    "CreateScheduleRequest",
     "CreateSpecialistPoolOperationMetadata",
     "CreateSpecialistPoolRequest",
     "CreateStudyRequest",
@@ -720,6 +736,7 @@
     "DeleteOperationMetadata",
     "DeletePipelineJobRequest",
     "DeleteSavedQueryRequest",
+    "DeleteScheduleRequest",
     "DeleteSpecialistPoolRequest",
     "DeleteStudyRequest",
     "DeleteTensorboardExperimentRequest",
@@ -820,6 +837,7 @@
     "GetNasTrialDetailRequest",
     "GetPipelineJobRequest",
     "GetPublisherModelRequest",
+    "GetScheduleRequest",
     "GetSpecialistPoolRequest",
     "GetStudyRequest",
     "GetTensorboardExperimentRequest",
@@ -908,6 +926,8 @@
     "ListPipelineJobsResponse",
     "ListSavedQueriesRequest",
     "ListSavedQueriesResponse",
+    "ListSchedulesRequest",
+    "ListSchedulesResponse",
     "ListSpecialistPoolsRequest",
     "ListSpecialistPoolsResponse",
     "ListStudiesRequest",
@@ -968,6 +988,7 @@
     "Neighbor",
     "NfsMount",
     "PauseModelDeploymentMonitoringJobRequest",
+    "PauseScheduleRequest",
     "PipelineFailurePolicy",
     "PipelineJob",
     "PipelineJobDetail",
@@ -1018,11 +1039,14 @@
     "RemoveDatapointsResponse",
     "ResourcesConsumed",
     "ResumeModelDeploymentMonitoringJobRequest",
+    "ResumeScheduleRequest",
     "SampleConfig",
     "SampledShapleyAttribution",
     "SamplingStrategy",
     "SavedQuery",
     "Scalar",
+    "Schedule",
+    "ScheduleServiceClient",
     "Scheduling",
     "SearchDataItemsRequest",
     "SearchDataItemsResponse",
@@ -1037,6 +1061,8 @@
     "SpecialistPoolServiceClient",
     "StopTrialRequest",
     "StratifiedSplit",
+    "StreamingPredictRequest",
+    "StreamingPredictResponse",
     "StreamingReadFeatureValuesRequest",
     "StringArray",
     "Study",
@@ -1045,6 +1071,7 @@
     "SuggestTrialsRequest",
     "SuggestTrialsResponse",
     "TFRecordDestination",
+    "Tensor",
     "Tensorboard",
     "TensorboardBlob",
     "TensorboardBlobSequence",
@@ -1085,6 +1112,7 @@
     "UpdateModelDeploymentMonitoringJobOperationMetadata",
     "UpdateModelDeploymentMonitoringJobRequest",
     "UpdateModelRequest",
+    "UpdateScheduleRequest",
     "UpdateSpecialistPoolOperationMetadata",
     "UpdateSpecialistPoolRequest",
     "UpdateTensorboardExperimentRequest",

google/cloud/aiplatform_v1/gapic_metadata.json

@@ -1806,6 +1806,11 @@
               "methods": [
                 "raw_predict"
               ]
+            },
+            "ServerStreamingPredict": {
+              "methods": [
+                "server_streaming_predict"
+              ]
             }
           }
         },
@@ -1826,6 +1831,95 @@
               "methods": [
                 "raw_predict"
               ]
+            },
+            "ServerStreamingPredict": {
+              "methods": [
+                "server_streaming_predict"
+              ]
+            }
+          }
+        }
+      }
+    },
+    "ScheduleService": {
+      "clients": {
+        "grpc": {
+          "libraryClient": "ScheduleServiceClient",
+          "rpcs": {
+            "CreateSchedule": {
+              "methods": [
+                "create_schedule"
+              ]
+            },
+            "DeleteSchedule": {
+              "methods": [
+                "delete_schedule"
+              ]
+            },
+            "GetSchedule": {
+              "methods": [
+                "get_schedule"
+              ]
+            },
+            "ListSchedules": {
+              "methods": [
+                "list_schedules"
+              ]
+            },
+            "PauseSchedule": {
+              "methods": [
+                "pause_schedule"
+              ]
+            },
+            "ResumeSchedule": {
+              "methods": [
+                "resume_schedule"
+              ]
+            },
+            "UpdateSchedule": {
+              "methods": [
+                "update_schedule"
+              ]
+            }
+          }
+        },
+        "grpc-async": {
+          "libraryClient": "ScheduleServiceAsyncClient",
+          "rpcs": {
+            "CreateSchedule": {
+              "methods": [
+                "create_schedule"
+              ]
+            },
+            "DeleteSchedule": {
+              "methods": [
+                "delete_schedule"
+              ]
+            },
+            "GetSchedule": {
+              "methods": [
+                "get_schedule"
+              ]
+            },
+            "ListSchedules": {
+              "methods": [
+                "list_schedules"
+              ]
+            },
+            "PauseSchedule": {
+              "methods": [
+                "pause_schedule"
+              ]
+            },
+            "ResumeSchedule": {
+              "methods": [
+                "resume_schedule"
+              ]
+            },
+            "UpdateSchedule": {
+              "methods": [
+                "update_schedule"
+              ]
             }
           }
         }

google/cloud/aiplatform_v1/services/migration_service/client.py

@@ -230,40 +230,40 @@ def parse_dataset_path(path: str) -> Dict[str, str]:
     @staticmethod
     def dataset_path(
         project: str,
+        location: str,
         dataset: str,
     ) -> str:
         """Returns a fully-qualified dataset string."""
-        return "projects/{project}/datasets/{dataset}".format(
+        return "projects/{project}/locations/{location}/datasets/{dataset}".format(
             project=project,
+            location=location,
             dataset=dataset,
         )
 
     @staticmethod
     def parse_dataset_path(path: str) -> Dict[str, str]:
         """Parses a dataset path into its component segments."""
-        m = re.match(r"^projects/(?P<project>.+?)/datasets/(?P<dataset>.+?)$", path)
+        m = re.match(
+            r"^projects/(?P<project>.+?)/locations/(?P<location>.+?)/datasets/(?P<dataset>.+?)$",
+            path,
+        )
         return m.groupdict() if m else {}
 
     @staticmethod
     def dataset_path(
         project: str,
-        location: str,
         dataset: str,
     ) -> str:
         """Returns a fully-qualified dataset string."""
-        return "projects/{project}/locations/{location}/datasets/{dataset}".format(
+        return "projects/{project}/datasets/{dataset}".format(
             project=project,
-            location=location,
             dataset=dataset,
         )
 
     @staticmethod
     def parse_dataset_path(path: str) -> Dict[str, str]:
         """Parses a dataset path into its component segments."""
-        m = re.match(
-            r"^projects/(?P<project>.+?)/locations/(?P<location>.+?)/datasets/(?P<dataset>.+?)$",
-            path,
-        )
+        m = re.match(r"^projects/(?P<project>.+?)/datasets/(?P<dataset>.+?)$", path)
         return m.groupdict() if m else {}
 
     @staticmethod