Skip to content

Commit 1a580ae

Browse files
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>
1 parent e87a255 commit 1a580ae

File tree

125 files changed

+10586
-784
lines changed

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

125 files changed

+10586
-784
lines changed

docs/aiplatform_v1/services.rst

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -13,3 +13,4 @@ Services for Google Cloud Aiplatform v1 API
1313
pipeline_service
1414
prediction_service
1515
specialist_pool_service
16+
vizier_service
Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,10 @@
1+
VizierService
2+
-------------------------------
3+
4+
.. automodule:: google.cloud.aiplatform_v1.services.vizier_service
5+
:members:
6+
:inherited-members:
7+
8+
.. automodule:: google.cloud.aiplatform_v1.services.vizier_service.pagers
9+
:members:
10+
:inherited-members:

google/cloud/aiplatform_v1/__init__.py

Lines changed: 48 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -34,6 +34,8 @@
3434
from .services.prediction_service import PredictionServiceAsyncClient
3535
from .services.specialist_pool_service import SpecialistPoolServiceClient
3636
from .services.specialist_pool_service import SpecialistPoolServiceAsyncClient
37+
from .services.vizier_service import VizierServiceClient
38+
from .services.vizier_service import VizierServiceAsyncClient
3739

3840
from .types.accelerator_type import AcceleratorType
3941
from .types.annotation import Annotation
@@ -279,6 +281,28 @@
279281
from .types.training_pipeline import TrainingPipeline
280282
from .types.user_action_reference import UserActionReference
281283
from .types.value import Value
284+
from .types.vizier_service import AddTrialMeasurementRequest
285+
from .types.vizier_service import CheckTrialEarlyStoppingStateMetatdata
286+
from .types.vizier_service import CheckTrialEarlyStoppingStateRequest
287+
from .types.vizier_service import CheckTrialEarlyStoppingStateResponse
288+
from .types.vizier_service import CompleteTrialRequest
289+
from .types.vizier_service import CreateStudyRequest
290+
from .types.vizier_service import CreateTrialRequest
291+
from .types.vizier_service import DeleteStudyRequest
292+
from .types.vizier_service import DeleteTrialRequest
293+
from .types.vizier_service import GetStudyRequest
294+
from .types.vizier_service import GetTrialRequest
295+
from .types.vizier_service import ListOptimalTrialsRequest
296+
from .types.vizier_service import ListOptimalTrialsResponse
297+
from .types.vizier_service import ListStudiesRequest
298+
from .types.vizier_service import ListStudiesResponse
299+
from .types.vizier_service import ListTrialsRequest
300+
from .types.vizier_service import ListTrialsResponse
301+
from .types.vizier_service import LookupStudyRequest
302+
from .types.vizier_service import StopTrialRequest
303+
from .types.vizier_service import SuggestTrialsMetadata
304+
from .types.vizier_service import SuggestTrialsRequest
305+
from .types.vizier_service import SuggestTrialsResponse
282306

283307
__all__ = (
284308
"DatasetServiceAsyncClient",
@@ -291,8 +315,10 @@
291315
"PipelineServiceAsyncClient",
292316
"PredictionServiceAsyncClient",
293317
"SpecialistPoolServiceAsyncClient",
318+
"VizierServiceAsyncClient",
294319
"AcceleratorType",
295320
"ActiveLearningConfig",
321+
"AddTrialMeasurementRequest",
296322
"Annotation",
297323
"AnnotationSpec",
298324
"Artifact",
@@ -312,6 +338,10 @@
312338
"CancelHyperparameterTuningJobRequest",
313339
"CancelPipelineJobRequest",
314340
"CancelTrainingPipelineRequest",
341+
"CheckTrialEarlyStoppingStateMetatdata",
342+
"CheckTrialEarlyStoppingStateRequest",
343+
"CheckTrialEarlyStoppingStateResponse",
344+
"CompleteTrialRequest",
315345
"CompletionStats",
316346
"ContainerRegistryDestination",
317347
"ContainerSpec",
@@ -332,7 +362,9 @@
332362
"CreatePipelineJobRequest",
333363
"CreateSpecialistPoolOperationMetadata",
334364
"CreateSpecialistPoolRequest",
365+
"CreateStudyRequest",
335366
"CreateTrainingPipelineRequest",
367+
"CreateTrialRequest",
336368
"CustomJob",
337369
"CustomJobSpec",
338370
"DataItem",
@@ -353,7 +385,9 @@
353385
"DeleteOperationMetadata",
354386
"DeletePipelineJobRequest",
355387
"DeleteSpecialistPoolRequest",
388+
"DeleteStudyRequest",
356389
"DeleteTrainingPipelineRequest",
390+
"DeleteTrialRequest",
357391
"DeployIndexOperationMetadata",
358392
"DeployIndexRequest",
359393
"DeployIndexResponse",
@@ -408,7 +442,9 @@
408442
"GetModelRequest",
409443
"GetPipelineJobRequest",
410444
"GetSpecialistPoolRequest",
445+
"GetStudyRequest",
411446
"GetTrainingPipelineRequest",
447+
"GetTrialRequest",
412448
"HyperparameterTuningJob",
413449
"ImportDataConfig",
414450
"ImportDataOperationMetadata",
@@ -451,12 +487,19 @@
451487
"ListModelEvaluationsResponse",
452488
"ListModelsRequest",
453489
"ListModelsResponse",
490+
"ListOptimalTrialsRequest",
491+
"ListOptimalTrialsResponse",
454492
"ListPipelineJobsRequest",
455493
"ListPipelineJobsResponse",
456494
"ListSpecialistPoolsRequest",
457495
"ListSpecialistPoolsResponse",
496+
"ListStudiesRequest",
497+
"ListStudiesResponse",
458498
"ListTrainingPipelinesRequest",
459499
"ListTrainingPipelinesResponse",
500+
"ListTrialsRequest",
501+
"ListTrialsResponse",
502+
"LookupStudyRequest",
460503
"MachineSpec",
461504
"ManualBatchTuningParameters",
462505
"Measurement",
@@ -507,8 +550,12 @@
507550
"SmoothGradConfig",
508551
"SpecialistPool",
509552
"SpecialistPoolServiceClient",
553+
"StopTrialRequest",
510554
"Study",
511555
"StudySpec",
556+
"SuggestTrialsMetadata",
557+
"SuggestTrialsRequest",
558+
"SuggestTrialsResponse",
512559
"ThresholdConfig",
513560
"TimestampSplit",
514561
"TrainingConfig",
@@ -535,6 +582,7 @@
535582
"UploadModelResponse",
536583
"UserActionReference",
537584
"Value",
585+
"VizierServiceClient",
538586
"WorkerPoolSpec",
539587
"XraiAttribution",
540588
)

google/cloud/aiplatform_v1/gapic_metadata.json

Lines changed: 164 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1014,6 +1014,170 @@
10141014
}
10151015
}
10161016
}
1017+
},
1018+
"VizierService": {
1019+
"clients": {
1020+
"grpc": {
1021+
"libraryClient": "VizierServiceClient",
1022+
"rpcs": {
1023+
"AddTrialMeasurement": {
1024+
"methods": [
1025+
"add_trial_measurement"
1026+
]
1027+
},
1028+
"CheckTrialEarlyStoppingState": {
1029+
"methods": [
1030+
"check_trial_early_stopping_state"
1031+
]
1032+
},
1033+
"CompleteTrial": {
1034+
"methods": [
1035+
"complete_trial"
1036+
]
1037+
},
1038+
"CreateStudy": {
1039+
"methods": [
1040+
"create_study"
1041+
]
1042+
},
1043+
"CreateTrial": {
1044+
"methods": [
1045+
"create_trial"
1046+
]
1047+
},
1048+
"DeleteStudy": {
1049+
"methods": [
1050+
"delete_study"
1051+
]
1052+
},
1053+
"DeleteTrial": {
1054+
"methods": [
1055+
"delete_trial"
1056+
]
1057+
},
1058+
"GetStudy": {
1059+
"methods": [
1060+
"get_study"
1061+
]
1062+
},
1063+
"GetTrial": {
1064+
"methods": [
1065+
"get_trial"
1066+
]
1067+
},
1068+
"ListOptimalTrials": {
1069+
"methods": [
1070+
"list_optimal_trials"
1071+
]
1072+
},
1073+
"ListStudies": {
1074+
"methods": [
1075+
"list_studies"
1076+
]
1077+
},
1078+
"ListTrials": {
1079+
"methods": [
1080+
"list_trials"
1081+
]
1082+
},
1083+
"LookupStudy": {
1084+
"methods": [
1085+
"lookup_study"
1086+
]
1087+
},
1088+
"StopTrial": {
1089+
"methods": [
1090+
"stop_trial"
1091+
]
1092+
},
1093+
"SuggestTrials": {
1094+
"methods": [
1095+
"suggest_trials"
1096+
]
1097+
}
1098+
}
1099+
},
1100+
"grpc-async": {
1101+
"libraryClient": "VizierServiceAsyncClient",
1102+
"rpcs": {
1103+
"AddTrialMeasurement": {
1104+
"methods": [
1105+
"add_trial_measurement"
1106+
]
1107+
},
1108+
"CheckTrialEarlyStoppingState": {
1109+
"methods": [
1110+
"check_trial_early_stopping_state"
1111+
]
1112+
},
1113+
"CompleteTrial": {
1114+
"methods": [
1115+
"complete_trial"
1116+
]
1117+
},
1118+
"CreateStudy": {
1119+
"methods": [
1120+
"create_study"
1121+
]
1122+
},
1123+
"CreateTrial": {
1124+
"methods": [
1125+
"create_trial"
1126+
]
1127+
},
1128+
"DeleteStudy": {
1129+
"methods": [
1130+
"delete_study"
1131+
]
1132+
},
1133+
"DeleteTrial": {
1134+
"methods": [
1135+
"delete_trial"
1136+
]
1137+
},
1138+
"GetStudy": {
1139+
"methods": [
1140+
"get_study"
1141+
]
1142+
},
1143+
"GetTrial": {
1144+
"methods": [
1145+
"get_trial"
1146+
]
1147+
},
1148+
"ListOptimalTrials": {
1149+
"methods": [
1150+
"list_optimal_trials"
1151+
]
1152+
},
1153+
"ListStudies": {
1154+
"methods": [
1155+
"list_studies"
1156+
]
1157+
},
1158+
"ListTrials": {
1159+
"methods": [
1160+
"list_trials"
1161+
]
1162+
},
1163+
"LookupStudy": {
1164+
"methods": [
1165+
"lookup_study"
1166+
]
1167+
},
1168+
"StopTrial": {
1169+
"methods": [
1170+
"stop_trial"
1171+
]
1172+
},
1173+
"SuggestTrials": {
1174+
"methods": [
1175+
"suggest_trials"
1176+
]
1177+
}
1178+
}
1179+
}
1180+
}
10171181
}
10181182
}
10191183
}

0 commit comments

Comments
 (0)