Skip to content

Commit 4ea5a27

Browse files
Google APIscopybara-github
Google APIs
authored and
copybara-github
committed
feat!: BREAKING CHANGE: remove unsupported accelerator types
feat: add aiplatform API Vizier service Committer: @dizcology PiperOrigin-RevId: 363921711
1 parent 7998de5 commit 4ea5a27

42 files changed

Lines changed: 1160 additions & 118 deletions

Some content is hidden

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

‎google/cloud/aiplatform/v1beta1/BUILD.bazel‎

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -59,6 +59,7 @@ proto_library(
5959
"study.proto",
6060
"training_pipeline.proto",
6161
"user_action_reference.proto",
62+
"vizier_service.proto"
6263
],
6364
deps = [
6465
"//google/api:annotations_proto",
@@ -145,6 +146,7 @@ java_gapic_test(
145146
"com.google.cloud.aiplatform.v1beta1.PipelineServiceClientTest",
146147
"com.google.cloud.aiplatform.v1beta1.PredictionServiceClientTest",
147148
"com.google.cloud.aiplatform.v1beta1.SpecialistPoolServiceClientTest",
149+
"com.google.cloud.aiplatform.v1beta1.VizierServiceClientTest",
148150
],
149151
runtime_deps = [":aiplatform_java_gapic_test"],
150152
)

‎google/cloud/aiplatform/v1beta1/accelerator_type.proto‎

Lines changed: 1 addition & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
// Copyright 2020 Google LLC
1+
// Copyright 2021 Google LLC
22
//
33
// Licensed under the Apache License, Version 2.0 (the "License");
44
// you may not use this file except in compliance with the License.
@@ -42,10 +42,4 @@ enum AcceleratorType {
4242

4343
// Nvidia Tesla T4 GPU.
4444
NVIDIA_TESLA_T4 = 5;
45-
46-
// TPU v2.
47-
TPU_V2 = 6;
48-
49-
// TPU v3.
50-
TPU_V3 = 7;
5145
}

‎google/cloud/aiplatform/v1beta1/aiplatform_grpc_service_config.json‎

Lines changed: 65 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -274,6 +274,71 @@
274274
}
275275
],
276276
"timeout": "5s"
277+
},
278+
{
279+
"name":[
280+
{
281+
"service": "google.cloud.aiplatform.v1beta1.VizierService",
282+
"method": "CreateStudy"
283+
},
284+
{
285+
"service": "google.cloud.aiplatform.v1beta1.VizierService",
286+
"method": "GetStudy"
287+
},
288+
{
289+
"service": "google.cloud.aiplatform.v1beta1.VizierService",
290+
"method": "ListStudies"
291+
},
292+
{
293+
"service": "google.cloud.aiplatform.v1beta1.VizierService",
294+
"method": "DeleteStudy"
295+
},
296+
{
297+
"service": "google.cloud.aiplatform.v1beta1.VizierService",
298+
"method": "LookupStudy"
299+
},
300+
{
301+
"service": "google.cloud.aiplatform.v1beta1.VizierService",
302+
"method": "SuggestTrials"
303+
},
304+
{
305+
"service": "google.cloud.aiplatform.v1beta1.VizierService",
306+
"method": "CreateTrial"
307+
},
308+
{
309+
"service": "google.cloud.aiplatform.v1beta1.VizierService",
310+
"method": "GetTrial"
311+
},
312+
{
313+
"service": "google.cloud.aiplatform.v1beta1.VizierService",
314+
"method": "ListTrials"
315+
},
316+
{
317+
"service": "google.cloud.aiplatform.v1beta1.VizierService",
318+
"method": "AddTrialMeasurement"
319+
},
320+
{
321+
"service": "google.cloud.aiplatform.v1beta1.VizierService",
322+
"method": "CompleteTrial"
323+
},
324+
{
325+
"service": "google.cloud.aiplatform.v1beta1.VizierService",
326+
"method": "DeleteTrial"
327+
},
328+
{
329+
"service": "google.cloud.aiplatform.v1beta1.VizierService",
330+
"method": "CheckTrialEarlyStoppingState"
331+
},
332+
{
333+
"service": "google.cloud.aiplatform.v1beta1.VizierService",
334+
"method": "StopTrial"
335+
},
336+
{
337+
"service": "google.cloud.aiplatform.v1beta1.VizierService",
338+
"method": "ListOptimalTrials"
339+
}
340+
],
341+
"timeout": "5s"
277342
}
278343
]
279344
}

‎google/cloud/aiplatform/v1beta1/aiplatform_v1beta1.yaml‎

Lines changed: 320 additions & 0 deletions
Large diffs are not rendered by default.

‎google/cloud/aiplatform/v1beta1/annotation.proto‎

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
// Copyright 2020 Google LLC
1+
// Copyright 2021 Google LLC
22
//
33
// Licensed under the Apache License, Version 2.0 (the "License");
44
// you may not use this file except in compliance with the License.
@@ -18,6 +18,7 @@ package google.cloud.aiplatform.v1beta1;
1818

1919
import "google/api/field_behavior.proto";
2020
import "google/api/resource.proto";
21+
2122
import "google/cloud/aiplatform/v1beta1/user_action_reference.proto";
2223
import "google/protobuf/struct.proto";
2324
import "google/protobuf/timestamp.proto";
@@ -58,7 +59,7 @@ message Annotation {
5859
// Output only. Timestamp when this Annotation was last updated.
5960
google.protobuf.Timestamp update_time = 7 [(google.api.field_behavior) = OUTPUT_ONLY];
6061

61-
// Optional. Used to perform a consistent read-modify-write updates. If not set, a blind
62+
// Optional. Used to perform consistent read-modify-write updates. If not set, a blind
6263
// "overwrite" update happens.
6364
string etag = 8 [(google.api.field_behavior) = OPTIONAL];
6465

@@ -79,7 +80,7 @@ message Annotation {
7980
//
8081
// * "aiplatform.googleapis.com/annotation_set_name":
8182
// optional, name of the UI's annotation set this Annotation belongs to.
82-
// If not set the Annotation is not visible in the UI.
83+
// If not set, the Annotation is not visible in the UI.
8384
//
8485
// * "aiplatform.googleapis.com/payload_schema":
8586
// output only, its value is the [payload_schema's][google.cloud.aiplatform.v1beta1.Annotation.payload_schema_uri]

‎google/cloud/aiplatform/v1beta1/annotation_spec.proto‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
// Copyright 2020 Google LLC
1+
// Copyright 2021 Google LLC
22
//
33
// Licensed under the Apache License, Version 2.0 (the "License");
44
// you may not use this file except in compliance with the License.
@@ -47,7 +47,7 @@ message AnnotationSpec {
4747
// Output only. Timestamp when AnnotationSpec was last updated.
4848
google.protobuf.Timestamp update_time = 4 [(google.api.field_behavior) = OUTPUT_ONLY];
4949

50-
// Optional. Used to perform a consistent read-modify-write updates. If not set, a blind
50+
// Optional. Used to perform consistent read-modify-write updates. If not set, a blind
5151
// "overwrite" update happens.
5252
string etag = 5 [(google.api.field_behavior) = OPTIONAL];
5353
}

‎google/cloud/aiplatform/v1beta1/batch_prediction_job.proto‎

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
// Copyright 2020 Google LLC
1+
// Copyright 2021 Google LLC
22
//
33
// Licensed under the Apache License, Version 2.0 (the "License");
44
// you may not use this file except in compliance with the License.
@@ -120,7 +120,6 @@ message BatchPredictionJob {
120120

121121
// Required. The format in which AI Platform gives the predictions, must be one of the
122122
// [Model's][google.cloud.aiplatform.v1beta1.BatchPredictionJob.model]
123-
//
124123
// [supported_output_storage_formats][google.cloud.aiplatform.v1beta1.Model.supported_output_storage_formats].
125124
string predictions_format = 1 [(google.api.field_behavior) = REQUIRED];
126125
}
@@ -206,8 +205,8 @@ message BatchPredictionJob {
206205
// conforms to the [Explanation][google.cloud.aiplatform.v1beta1.Explanation] object.
207206
// * `csv`: Generating explanations for CSV format is not supported.
208207
//
209-
// If this field is set to true, the [Model.explanation_spec][google.cloud.aiplatform.v1beta1.Model.explanation_spec] must be
210-
// populated.
208+
// If this field is set to true, either the [Model.explanation_spec][google.cloud.aiplatform.v1beta1.Model.explanation_spec] or
209+
// [explanation_spec][google.cloud.aiplatform.v1beta1.BatchPredictionJob.explanation_spec] must be populated.
211210
bool generate_explanation = 23;
212211

213212
// Explanation configuration for this BatchPredictionJob. Can be

‎google/cloud/aiplatform/v1beta1/completion_stats.proto‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
// Copyright 2020 Google LLC
1+
// Copyright 2021 Google LLC
22
//
33
// Licensed under the Apache License, Version 2.0 (the "License");
44
// you may not use this file except in compliance with the License.

‎google/cloud/aiplatform/v1beta1/custom_job.proto‎

Lines changed: 9 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
// Copyright 2020 Google LLC
1+
// Copyright 2021 Google LLC
22
//
33
// Licensed under the Apache License, Version 2.0 (the "License");
44
// you may not use this file except in compliance with the License.
@@ -93,6 +93,8 @@ message CustomJob {
9393
// Represents the spec of a CustomJob.
9494
message CustomJobSpec {
9595
// Required. The spec of the worker pools including machine type and Docker image.
96+
// All worker pools except the first one are optional and can be skipped by
97+
// providing an empty value.
9698
repeated WorkerPoolSpec worker_pool_specs = 1 [(google.api.field_behavior) = REQUIRED];
9799

98100
// Scheduling options for a CustomJob.
@@ -180,10 +182,12 @@ message ContainerSpec {
180182

181183
// The spec of a Python packaged code.
182184
message PythonPackageSpec {
183-
// Required. The URI of a container image in the Container Registry that will run the
184-
// provided python package. AI Platform provides wide range of executor images
185-
// with pre-installed packages to meet users' various use cases. Only one of
186-
// the provided images can be set here.
185+
// Required. The URI of a container image in Artifact Registry that will run the
186+
// provided Python package. AI Platform provides a wide range of executor
187+
// images with pre-installed packages to meet users' various use cases. See
188+
// the list of [pre-built containers for
189+
// training](https://cloud.google.com/ai-platform-unified/docs/training/pre-built-containers).
190+
// You must use an image from this list.
187191
string executor_image_uri = 1 [(google.api.field_behavior) = REQUIRED];
188192

189193
// Required. The Google Cloud Storage location of the Python package files which are

‎google/cloud/aiplatform/v1beta1/data_item.proto‎

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
// Copyright 2020 Google LLC
1+
// Copyright 2021 Google LLC
22
//
33
// Licensed under the Apache License, Version 2.0 (the "License");
44
// you may not use this file except in compliance with the License.
@@ -18,6 +18,7 @@ package google.cloud.aiplatform.v1beta1;
1818

1919
import "google/api/field_behavior.proto";
2020
import "google/api/resource.proto";
21+
2122
import "google/protobuf/struct.proto";
2223
import "google/protobuf/timestamp.proto";
2324
import "google/api/annotations.proto";
@@ -62,7 +63,7 @@ message DataItem {
6263
// [metadata schema's][google.cloud.aiplatform.v1beta1.Dataset.metadata_schema_uri] dataItemSchemaUri field.
6364
google.protobuf.Value payload = 4 [(google.api.field_behavior) = REQUIRED];
6465

65-
// Optional. Used to perform a consistent read-modify-write updates. If not set, a blind
66+
// Optional. Used to perform consistent read-modify-write updates. If not set, a blind
6667
// "overwrite" update happens.
6768
string etag = 7 [(google.api.field_behavior) = OPTIONAL];
6869
}

0 commit comments

Comments
 (0)