feat: add vizier service to aiplatform v1 BUILD.bazel (#731)

* feat: add vizier service to aiplatform v1 BUILD.bazel Committer: @dizcology PiperOrigin-RevId: 399538263 Source-Link: googleapis/googleapis@70d5a82 Source-Link: googleapis/googleapis-gen@eef2032 Copy-Tag: eyJwIjoiLmdpdGh1Yi8uT3dsQm90LnlhbWwiLCJoIjoiZWVmMjAzMjQxNzA0ZTQzMDFlOTYzYTZkMjFhZGRlNGY2YjMyYWMwZSJ9 * 🦉 Updates from OwlBot See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md Co-authored-by: Owl Bot <gcf-owl-bot[bot]@users.noreply.github.com>
googleapis · Sep 29, 2021 · 1a580ae · 1a580ae
1 parent e87a255
commit 1a580ae
Show file tree

Hide file tree

Showing 125 changed files with 10,586 additions and 784 deletions.
diff --git a/docs/aiplatform_v1/services.rst b/docs/aiplatform_v1/services.rst
@@ -13,3 +13,4 @@ Services for Google Cloud Aiplatform v1 API
     pipeline_service
     prediction_service
     specialist_pool_service
+    vizier_service
diff --git a/docs/aiplatform_v1/vizier_service.rst b/docs/aiplatform_v1/vizier_service.rst
@@ -0,0 +1,10 @@
+VizierService
+-------------------------------
+
+.. automodule:: google.cloud.aiplatform_v1.services.vizier_service
+    :members:
+    :inherited-members:
+
+.. automodule:: google.cloud.aiplatform_v1.services.vizier_service.pagers
+    :members:
+    :inherited-members:
diff --git a/google/cloud/aiplatform_v1/__init__.py b/google/cloud/aiplatform_v1/__init__.py
@@ -34,6 +34,8 @@
 from .services.prediction_service import PredictionServiceAsyncClient
 from .services.specialist_pool_service import SpecialistPoolServiceClient
 from .services.specialist_pool_service import SpecialistPoolServiceAsyncClient
+from .services.vizier_service import VizierServiceClient
+from .services.vizier_service import VizierServiceAsyncClient
 
 from .types.accelerator_type import AcceleratorType
 from .types.annotation import Annotation
@@ -279,6 +281,28 @@
 from .types.training_pipeline import TrainingPipeline
 from .types.user_action_reference import UserActionReference
 from .types.value import Value
+from .types.vizier_service import AddTrialMeasurementRequest
+from .types.vizier_service import CheckTrialEarlyStoppingStateMetatdata
+from .types.vizier_service import CheckTrialEarlyStoppingStateRequest
+from .types.vizier_service import CheckTrialEarlyStoppingStateResponse
+from .types.vizier_service import CompleteTrialRequest
+from .types.vizier_service import CreateStudyRequest
+from .types.vizier_service import CreateTrialRequest
+from .types.vizier_service import DeleteStudyRequest
+from .types.vizier_service import DeleteTrialRequest
+from .types.vizier_service import GetStudyRequest
+from .types.vizier_service import GetTrialRequest
+from .types.vizier_service import ListOptimalTrialsRequest
+from .types.vizier_service import ListOptimalTrialsResponse
+from .types.vizier_service import ListStudiesRequest
+from .types.vizier_service import ListStudiesResponse
+from .types.vizier_service import ListTrialsRequest
+from .types.vizier_service import ListTrialsResponse
+from .types.vizier_service import LookupStudyRequest
+from .types.vizier_service import StopTrialRequest
+from .types.vizier_service import SuggestTrialsMetadata
+from .types.vizier_service import SuggestTrialsRequest
+from .types.vizier_service import SuggestTrialsResponse
 
 __all__ = (
     "DatasetServiceAsyncClient",
@@ -291,8 +315,10 @@
     "PipelineServiceAsyncClient",
     "PredictionServiceAsyncClient",
     "SpecialistPoolServiceAsyncClient",
+    "VizierServiceAsyncClient",
     "AcceleratorType",
     "ActiveLearningConfig",
+    "AddTrialMeasurementRequest",
     "Annotation",
     "AnnotationSpec",
     "Artifact",
@@ -312,6 +338,10 @@
     "CancelHyperparameterTuningJobRequest",
     "CancelPipelineJobRequest",
     "CancelTrainingPipelineRequest",
+    "CheckTrialEarlyStoppingStateMetatdata",
+    "CheckTrialEarlyStoppingStateRequest",
+    "CheckTrialEarlyStoppingStateResponse",
+    "CompleteTrialRequest",
     "CompletionStats",
     "ContainerRegistryDestination",
     "ContainerSpec",
@@ -332,7 +362,9 @@
     "CreatePipelineJobRequest",
     "CreateSpecialistPoolOperationMetadata",
     "CreateSpecialistPoolRequest",
+    "CreateStudyRequest",
     "CreateTrainingPipelineRequest",
+    "CreateTrialRequest",
     "CustomJob",
     "CustomJobSpec",
     "DataItem",
@@ -353,7 +385,9 @@
     "DeleteOperationMetadata",
     "DeletePipelineJobRequest",
     "DeleteSpecialistPoolRequest",
+    "DeleteStudyRequest",
     "DeleteTrainingPipelineRequest",
+    "DeleteTrialRequest",
     "DeployIndexOperationMetadata",
     "DeployIndexRequest",
     "DeployIndexResponse",
@@ -408,7 +442,9 @@
     "GetModelRequest",
     "GetPipelineJobRequest",
     "GetSpecialistPoolRequest",
+    "GetStudyRequest",
     "GetTrainingPipelineRequest",
+    "GetTrialRequest",
     "HyperparameterTuningJob",
     "ImportDataConfig",
     "ImportDataOperationMetadata",
@@ -451,12 +487,19 @@
     "ListModelEvaluationsResponse",
     "ListModelsRequest",
     "ListModelsResponse",
+    "ListOptimalTrialsRequest",
+    "ListOptimalTrialsResponse",
     "ListPipelineJobsRequest",
     "ListPipelineJobsResponse",
     "ListSpecialistPoolsRequest",
     "ListSpecialistPoolsResponse",
+    "ListStudiesRequest",
+    "ListStudiesResponse",
     "ListTrainingPipelinesRequest",
     "ListTrainingPipelinesResponse",
+    "ListTrialsRequest",
+    "ListTrialsResponse",
+    "LookupStudyRequest",
     "MachineSpec",
     "ManualBatchTuningParameters",
     "Measurement",
@@ -507,8 +550,12 @@
     "SmoothGradConfig",
     "SpecialistPool",
     "SpecialistPoolServiceClient",
+    "StopTrialRequest",
     "Study",
     "StudySpec",
+    "SuggestTrialsMetadata",
+    "SuggestTrialsRequest",
+    "SuggestTrialsResponse",
     "ThresholdConfig",
     "TimestampSplit",
     "TrainingConfig",
@@ -535,6 +582,7 @@
     "UploadModelResponse",
     "UserActionReference",
     "Value",
+    "VizierServiceClient",
     "WorkerPoolSpec",
     "XraiAttribution",
 )
diff --git a/google/cloud/aiplatform_v1/gapic_metadata.json b/google/cloud/aiplatform_v1/gapic_metadata.json
@@ -1014,6 +1014,170 @@
           }
         }
       }
+    },
+    "VizierService": {
+      "clients": {
+        "grpc": {
+          "libraryClient": "VizierServiceClient",
+          "rpcs": {
+            "AddTrialMeasurement": {
+              "methods": [
+                "add_trial_measurement"
+              ]
+            },
+            "CheckTrialEarlyStoppingState": {
+              "methods": [
+                "check_trial_early_stopping_state"
+              ]
+            },
+            "CompleteTrial": {
+              "methods": [
+                "complete_trial"
+              ]
+            },
+            "CreateStudy": {
+              "methods": [
+                "create_study"
+              ]
+            },
+            "CreateTrial": {
+              "methods": [
+                "create_trial"
+              ]
+            },
+            "DeleteStudy": {
+              "methods": [
+                "delete_study"
+              ]
+            },
+            "DeleteTrial": {
+              "methods": [
+                "delete_trial"
+              ]
+            },
+            "GetStudy": {
+              "methods": [
+                "get_study"
+              ]
+            },
+            "GetTrial": {
+              "methods": [
+                "get_trial"
+              ]
+            },
+            "ListOptimalTrials": {
+              "methods": [
+                "list_optimal_trials"
+              ]
+            },
+            "ListStudies": {
+              "methods": [
+                "list_studies"
+              ]
+            },
+            "ListTrials": {
+              "methods": [
+                "list_trials"
+              ]
+            },
+            "LookupStudy": {
+              "methods": [
+                "lookup_study"
+              ]
+            },
+            "StopTrial": {
+              "methods": [
+                "stop_trial"
+              ]
+            },
+            "SuggestTrials": {
+              "methods": [
+                "suggest_trials"
+              ]
+            }
+          }
+        },
+        "grpc-async": {
+          "libraryClient": "VizierServiceAsyncClient",
+          "rpcs": {
+            "AddTrialMeasurement": {
+              "methods": [
+                "add_trial_measurement"
+              ]
+            },
+            "CheckTrialEarlyStoppingState": {
+              "methods": [
+                "check_trial_early_stopping_state"
+              ]
+            },
+            "CompleteTrial": {
+              "methods": [
+                "complete_trial"
+              ]
+            },
+            "CreateStudy": {
+              "methods": [
+                "create_study"
+              ]
+            },
+            "CreateTrial": {
+              "methods": [
+                "create_trial"
+              ]
+            },
+            "DeleteStudy": {
+              "methods": [
+                "delete_study"
+              ]
+            },
+            "DeleteTrial": {
+              "methods": [
+                "delete_trial"
+              ]
+            },
+            "GetStudy": {
+              "methods": [
+                "get_study"
+              ]
+            },
+            "GetTrial": {
+              "methods": [
+                "get_trial"
+              ]
+            },
+            "ListOptimalTrials": {
+              "methods": [
+                "list_optimal_trials"
+              ]
+            },
+            "ListStudies": {
+              "methods": [
+                "list_studies"
+              ]
+            },
+            "ListTrials": {
+              "methods": [
+                "list_trials"
+              ]
+            },
+            "LookupStudy": {
+              "methods": [
+                "lookup_study"
+              ]
+            },
+            "StopTrial": {
+              "methods": [
+                "stop_trial"
+              ]
+            },
+            "SuggestTrials": {
+              "methods": [
+                "suggest_trials"
+              ]
+            }
+          }
+        }
+      }
     }
   }
 }