Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

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

Merged
merged 2 commits into from Sep 29, 2021
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Jump to
Jump to file
Failed to load files.
Diff view
Diff view
1 change: 1 addition & 0 deletions docs/aiplatform_v1/services.rst
Expand Up @@ -13,3 +13,4 @@ Services for Google Cloud Aiplatform v1 API
pipeline_service
prediction_service
specialist_pool_service
vizier_service
10 changes: 10 additions & 0 deletions 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:
48 changes: 48 additions & 0 deletions google/cloud/aiplatform_v1/__init__.py
Expand Up @@ -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
Expand Down Expand Up @@ -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",
Expand All @@ -291,8 +315,10 @@
"PipelineServiceAsyncClient",
"PredictionServiceAsyncClient",
"SpecialistPoolServiceAsyncClient",
"VizierServiceAsyncClient",
"AcceleratorType",
"ActiveLearningConfig",
"AddTrialMeasurementRequest",
"Annotation",
"AnnotationSpec",
"Artifact",
Expand All @@ -312,6 +338,10 @@
"CancelHyperparameterTuningJobRequest",
"CancelPipelineJobRequest",
"CancelTrainingPipelineRequest",
"CheckTrialEarlyStoppingStateMetatdata",
"CheckTrialEarlyStoppingStateRequest",
"CheckTrialEarlyStoppingStateResponse",
"CompleteTrialRequest",
"CompletionStats",
"ContainerRegistryDestination",
"ContainerSpec",
Expand All @@ -332,7 +362,9 @@
"CreatePipelineJobRequest",
"CreateSpecialistPoolOperationMetadata",
"CreateSpecialistPoolRequest",
"CreateStudyRequest",
"CreateTrainingPipelineRequest",
"CreateTrialRequest",
"CustomJob",
"CustomJobSpec",
"DataItem",
Expand All @@ -353,7 +385,9 @@
"DeleteOperationMetadata",
"DeletePipelineJobRequest",
"DeleteSpecialistPoolRequest",
"DeleteStudyRequest",
"DeleteTrainingPipelineRequest",
"DeleteTrialRequest",
"DeployIndexOperationMetadata",
"DeployIndexRequest",
"DeployIndexResponse",
Expand Down Expand Up @@ -408,7 +442,9 @@
"GetModelRequest",
"GetPipelineJobRequest",
"GetSpecialistPoolRequest",
"GetStudyRequest",
"GetTrainingPipelineRequest",
"GetTrialRequest",
"HyperparameterTuningJob",
"ImportDataConfig",
"ImportDataOperationMetadata",
Expand Down Expand Up @@ -451,12 +487,19 @@
"ListModelEvaluationsResponse",
"ListModelsRequest",
"ListModelsResponse",
"ListOptimalTrialsRequest",
"ListOptimalTrialsResponse",
"ListPipelineJobsRequest",
"ListPipelineJobsResponse",
"ListSpecialistPoolsRequest",
"ListSpecialistPoolsResponse",
"ListStudiesRequest",
"ListStudiesResponse",
"ListTrainingPipelinesRequest",
"ListTrainingPipelinesResponse",
"ListTrialsRequest",
"ListTrialsResponse",
"LookupStudyRequest",
"MachineSpec",
"ManualBatchTuningParameters",
"Measurement",
Expand Down Expand Up @@ -507,8 +550,12 @@
"SmoothGradConfig",
"SpecialistPool",
"SpecialistPoolServiceClient",
"StopTrialRequest",
"Study",
"StudySpec",
"SuggestTrialsMetadata",
"SuggestTrialsRequest",
"SuggestTrialsResponse",
"ThresholdConfig",
"TimestampSplit",
"TrainingConfig",
Expand All @@ -535,6 +582,7 @@
"UploadModelResponse",
"UserActionReference",
"Value",
"VizierServiceClient",
"WorkerPoolSpec",
"XraiAttribution",
)
164 changes: 164 additions & 0 deletions google/cloud/aiplatform_v1/gapic_metadata.json
Expand Up @@ -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"
]
}
}
}
}
}
}
}