feat: Removes AcceleratorType.TPU_V2 and TPU_V3 constants (#543)

PiperOrigin-RevId: 384714314

Source-Link: googleapis/googleapis@cc51e48

Source-Link: googleapis/googleapis-gen@f87b91c

feat: Adds AcceleratorType.NVIDIA_TESLA_A100 constant
feat: Adds BigQuery output table field to batch prediction job output config 
feat: Adds JobState.JOB_STATE_EXPIRED constant 
feat: Adds AutoscalingMetricSpec message 
feat: Adds PipelineService methods for Create, Get, List, Delete, Cancel 
feat: Adds fields to Study message

google/cloud/aiplatform_v1/__init__.py

@@ -34,8 +34,10 @@
 from .types.accelerator_type import AcceleratorType
 from .types.annotation import Annotation
 from .types.annotation_spec import AnnotationSpec
+from .types.artifact import Artifact
 from .types.batch_prediction_job import BatchPredictionJob
 from .types.completion_stats import CompletionStats
+from .types.context import Context
 from .types.custom_job import ContainerSpec
 from .types.custom_job import CustomJob
 from .types.custom_job import CustomJobSpec
@@ -86,6 +88,7 @@
 from .types.endpoint_service import UndeployModelResponse
 from .types.endpoint_service import UpdateEndpointRequest
 from .types.env_var import EnvVar
+from .types.execution import Execution
 from .types.hyperparameter_tuning_job import HyperparameterTuningJob
 from .types.io import BigQueryDestination
 from .types.io import BigQuerySource
@@ -118,6 +121,7 @@
 from .types.job_service import ListHyperparameterTuningJobsResponse
 from .types.job_state import JobState
 from .types.machine_resources import AutomaticResources
+from .types.machine_resources import AutoscalingMetricSpec
 from .types.machine_resources import BatchDedicatedResources
 from .types.machine_resources import DedicatedResources
 from .types.machine_resources import DiskSpec
@@ -157,10 +161,20 @@
 from .types.model_service import UploadModelResponse
 from .types.operation import DeleteOperationMetadata
 from .types.operation import GenericOperationMetadata
+from .types.pipeline_job import PipelineJob
+from .types.pipeline_job import PipelineJobDetail
+from .types.pipeline_job import PipelineTaskDetail
+from .types.pipeline_job import PipelineTaskExecutorDetail
+from .types.pipeline_service import CancelPipelineJobRequest
 from .types.pipeline_service import CancelTrainingPipelineRequest
+from .types.pipeline_service import CreatePipelineJobRequest
 from .types.pipeline_service import CreateTrainingPipelineRequest
+from .types.pipeline_service import DeletePipelineJobRequest
 from .types.pipeline_service import DeleteTrainingPipelineRequest
+from .types.pipeline_service import GetPipelineJobRequest
 from .types.pipeline_service import GetTrainingPipelineRequest
+from .types.pipeline_service import ListPipelineJobsRequest
+from .types.pipeline_service import ListPipelineJobsResponse
 from .types.pipeline_service import ListTrainingPipelinesRequest
 from .types.pipeline_service import ListTrainingPipelinesResponse
 from .types.pipeline_state import PipelineState
@@ -185,6 +199,7 @@
 from .types.training_pipeline import TimestampSplit
 from .types.training_pipeline import TrainingPipeline
 from .types.user_action_reference import UserActionReference
+from .types.value import Value
 
 __all__ = (
     "DatasetServiceAsyncClient",
@@ -199,7 +214,9 @@
     "ActiveLearningConfig",
     "Annotation",
     "AnnotationSpec",
+    "Artifact",
     "AutomaticResources",
+    "AutoscalingMetricSpec",
     "BatchDedicatedResources",
     "BatchMigrateResourcesOperationMetadata",
     "BatchMigrateResourcesRequest",
@@ -211,10 +228,12 @@
     "CancelCustomJobRequest",
     "CancelDataLabelingJobRequest",
     "CancelHyperparameterTuningJobRequest",
+    "CancelPipelineJobRequest",
     "CancelTrainingPipelineRequest",
     "CompletionStats",
     "ContainerRegistryDestination",
     "ContainerSpec",
+    "Context",
     "CreateBatchPredictionJobRequest",
     "CreateCustomJobRequest",
     "CreateDataLabelingJobRequest",
@@ -223,6 +242,7 @@
     "CreateEndpointOperationMetadata",
     "CreateEndpointRequest",
     "CreateHyperparameterTuningJobRequest",
+    "CreatePipelineJobRequest",
     "CreateSpecialistPoolOperationMetadata",
     "CreateSpecialistPoolRequest",
     "CreateTrainingPipelineRequest",
@@ -241,6 +261,7 @@
     "DeleteHyperparameterTuningJobRequest",
     "DeleteModelRequest",
     "DeleteOperationMetadata",
+    "DeletePipelineJobRequest",
     "DeleteSpecialistPoolRequest",
     "DeleteTrainingPipelineRequest",
     "DeployModelOperationMetadata",
@@ -253,6 +274,7 @@
     "Endpoint",
     "EndpointServiceClient",
     "EnvVar",
+    "Execution",
     "ExportDataConfig",
     "ExportDataOperationMetadata",
     "ExportDataRequest",
@@ -275,6 +297,7 @@
     "GetModelEvaluationRequest",
     "GetModelEvaluationSliceRequest",
     "GetModelRequest",
+    "GetPipelineJobRequest",
     "GetSpecialistPoolRequest",
     "GetTrainingPipelineRequest",
     "HyperparameterTuningJob",
@@ -307,6 +330,8 @@
     "ListModelEvaluationsResponse",
     "ListModelsRequest",
     "ListModelsResponse",
+    "ListPipelineJobsRequest",
+    "ListPipelineJobsResponse",
     "ListSpecialistPoolsRequest",
     "ListSpecialistPoolsResponse",
     "ListTrainingPipelinesRequest",
@@ -323,8 +348,12 @@
     "ModelEvaluation",
     "ModelEvaluationSlice",
     "ModelServiceClient",
+    "PipelineJob",
+    "PipelineJobDetail",
     "PipelineServiceClient",
     "PipelineState",
+    "PipelineTaskDetail",
+    "PipelineTaskExecutorDetail",
     "Port",
     "PredefinedSplit",
     "PredictRequest",
@@ -356,5 +385,6 @@
     "UploadModelRequest",
     "UploadModelResponse",
     "UserActionReference",
+    "Value",
     "WorkerPoolSpec",
 )

google/cloud/aiplatform_v1/gapic_metadata.json

@@ -570,26 +570,51 @@
         "grpc": {
           "libraryClient": "PipelineServiceClient",
           "rpcs": {
+            "CancelPipelineJob": {
+              "methods": [
+                "cancel_pipeline_job"
+              ]
+            },
             "CancelTrainingPipeline": {
               "methods": [
                 "cancel_training_pipeline"
               ]
             },
+            "CreatePipelineJob": {
+              "methods": [
+                "create_pipeline_job"
+              ]
+            },
             "CreateTrainingPipeline": {
               "methods": [
                 "create_training_pipeline"
               ]
             },
+            "DeletePipelineJob": {
+              "methods": [
+                "delete_pipeline_job"
+              ]
+            },
             "DeleteTrainingPipeline": {
               "methods": [
                 "delete_training_pipeline"
               ]
             },
+            "GetPipelineJob": {
+              "methods": [
+                "get_pipeline_job"
+              ]
+            },
             "GetTrainingPipeline": {
               "methods": [
                 "get_training_pipeline"
               ]
             },
+            "ListPipelineJobs": {
+              "methods": [
+                "list_pipeline_jobs"
+              ]
+            },
             "ListTrainingPipelines": {
               "methods": [
                 "list_training_pipelines"
@@ -600,26 +625,51 @@
         "grpc-async": {
           "libraryClient": "PipelineServiceAsyncClient",
           "rpcs": {
+            "CancelPipelineJob": {
+              "methods": [
+                "cancel_pipeline_job"
+              ]
+            },
             "CancelTrainingPipeline": {
               "methods": [
                 "cancel_training_pipeline"
               ]
             },
+            "CreatePipelineJob": {
+              "methods": [
+                "create_pipeline_job"
+              ]
+            },
             "CreateTrainingPipeline": {
               "methods": [
                 "create_training_pipeline"
               ]
             },
+            "DeletePipelineJob": {
+              "methods": [
+                "delete_pipeline_job"
+              ]
+            },
             "DeleteTrainingPipeline": {
               "methods": [
                 "delete_training_pipeline"
               ]
             },
+            "GetPipelineJob": {
+              "methods": [
+                "get_pipeline_job"
+              ]
+            },
             "GetTrainingPipeline": {
               "methods": [
                 "get_training_pipeline"
               ]
             },
+            "ListPipelineJobs": {
+              "methods": [
+                "list_pipeline_jobs"
+              ]
+            },
             "ListTrainingPipelines": {
               "methods": [
                 "list_training_pipelines"

google/cloud/aiplatform_v1/services/job_service/async_client.py

@@ -87,6 +87,8 @@ class JobServiceAsyncClient:
     )
     model_path = staticmethod(JobServiceClient.model_path)
     parse_model_path = staticmethod(JobServiceClient.parse_model_path)
+    network_path = staticmethod(JobServiceClient.network_path)
+    parse_network_path = staticmethod(JobServiceClient.parse_network_path)
     trial_path = staticmethod(JobServiceClient.trial_path)
     parse_trial_path = staticmethod(JobServiceClient.parse_trial_path)
     common_billing_account_path = staticmethod(

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

@@ -285,6 +285,21 @@ def parse_model_path(path: str) -> Dict[str, str]:
         )
         return m.groupdict() if m else {}
 
+    @staticmethod
+    def network_path(project: str, network: str,) -> str:
+        """Returns a fully-qualified network string."""
+        return "projects/{project}/global/networks/{network}".format(
+            project=project, network=network,
+        )
+
+    @staticmethod
+    def parse_network_path(path: str) -> Dict[str, str]:
+        """Parses a network path into its component segments."""
+        m = re.match(
+            r"^projects/(?P<project>.+?)/global/networks/(?P<network>.+?)$", path
+        )
+        return m.groupdict() if m else {}
+
     @staticmethod
     def trial_path(project: str, location: str, study: str, trial: str,) -> str:
         """Returns a fully-qualified trial string."""