Skip to content

Commit 726b620

Browse files
gcf-owl-bot[bot]sasha-gitg
gcf-owl-bot[bot]
and
sasha-gitg
authored
feat: Adds support for google.protobuf.Value pipeline parameters in the parameter_values field (#808)
* feat: Adds support for `google.protobuf.Value` pipeline parameters in the `parameter_values` field PiperOrigin-RevId: 406492721 Source-Link: googleapis/googleapis@7a3953f Source-Link: googleapis/googleapis-gen@f055ce9 Copy-Tag: eyJwIjoiLmdpdGh1Yi8uT3dsQm90LnlhbWwiLCJoIjoiZjA1NWNlOWM4MzY5NjcxNWU5M2IzY2IyOWI4MTI0NmY0M2UwYTRiOCJ9 * 🦉 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> Co-authored-by: sasha-gitg <44654632+sasha-gitg@users.noreply.github.com>
1 parent 87cfaf6 commit 726b620

File tree

236 files changed

+2754
-4924
lines changed

Some content is hidden

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

236 files changed

+2754
-4924
lines changed

‎google/cloud/aiplatform/v1/schema/trainingjob/definition_v1/types/automl_tables.py‎

Lines changed: 24 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -43,15 +43,24 @@ class AutoMlTables(proto.Message):
4343
class AutoMlTablesInputs(proto.Message):
4444
r"""
4545
46+
This message has `oneof`_ fields (mutually exclusive fields).
47+
For each oneof, at most one member field can be set at the same time.
48+
Setting any member of the oneof automatically clears all other
49+
members.
50+
51+
.. _oneof: https://proto-plus-python.readthedocs.io/en/stable/fields.html#oneofs-mutually-exclusive-fields
52+
4653
Attributes:
4754
optimization_objective_recall_value (float):
4855
Required when optimization_objective is
4956
"maximize-precision-at-recall". Must be between 0 and 1,
5057
inclusive.
58+
This field is a member of `oneof`_ ``additional_optimization_objective_config``.
5159
optimization_objective_precision_value (float):
5260
Required when optimization_objective is
5361
"maximize-recall-at-precision". Must be between 0 and 1,
5462
inclusive.
63+
This field is a member of `oneof`_ ``additional_optimization_objective_config``.
5564
prediction_type (str):
5665
The type of prediction the Model is to
5766
produce. "classification" - Predict one out of
@@ -149,23 +158,38 @@ class AutoMlTablesInputs(proto.Message):
149158
class Transformation(proto.Message):
150159
r"""
151160
161+
This message has `oneof`_ fields (mutually exclusive fields).
162+
For each oneof, at most one member field can be set at the same time.
163+
Setting any member of the oneof automatically clears all other
164+
members.
165+
166+
.. _oneof: https://proto-plus-python.readthedocs.io/en/stable/fields.html#oneofs-mutually-exclusive-fields
167+
152168
Attributes:
153169
auto (google.cloud.aiplatform.v1.schema.trainingjob.definition_v1.types.AutoMlTablesInputs.Transformation.AutoTransformation):
154170
171+
This field is a member of `oneof`_ ``transformation_detail``.
155172
numeric (google.cloud.aiplatform.v1.schema.trainingjob.definition_v1.types.AutoMlTablesInputs.Transformation.NumericTransformation):
156173
174+
This field is a member of `oneof`_ ``transformation_detail``.
157175
categorical (google.cloud.aiplatform.v1.schema.trainingjob.definition_v1.types.AutoMlTablesInputs.Transformation.CategoricalTransformation):
158176
177+
This field is a member of `oneof`_ ``transformation_detail``.
159178
timestamp (google.cloud.aiplatform.v1.schema.trainingjob.definition_v1.types.AutoMlTablesInputs.Transformation.TimestampTransformation):
160179
180+
This field is a member of `oneof`_ ``transformation_detail``.
161181
text (google.cloud.aiplatform.v1.schema.trainingjob.definition_v1.types.AutoMlTablesInputs.Transformation.TextTransformation):
162182
183+
This field is a member of `oneof`_ ``transformation_detail``.
163184
repeated_numeric (google.cloud.aiplatform.v1.schema.trainingjob.definition_v1.types.AutoMlTablesInputs.Transformation.NumericArrayTransformation):
164185
186+
This field is a member of `oneof`_ ``transformation_detail``.
165187
repeated_categorical (google.cloud.aiplatform.v1.schema.trainingjob.definition_v1.types.AutoMlTablesInputs.Transformation.CategoricalArrayTransformation):
166188
189+
This field is a member of `oneof`_ ``transformation_detail``.
167190
repeated_text (google.cloud.aiplatform.v1.schema.trainingjob.definition_v1.types.AutoMlTablesInputs.Transformation.TextArrayTransformation):
168191
192+
This field is a member of `oneof`_ ``transformation_detail``.
169193
"""
170194

171195
class AutoTransformation(proto.Message):

‎google/cloud/aiplatform/v1beta1/schema/trainingjob/definition_v1beta1/types/automl_tables.py‎

Lines changed: 24 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -43,15 +43,24 @@ class AutoMlTables(proto.Message):
4343
class AutoMlTablesInputs(proto.Message):
4444
r"""
4545
46+
This message has `oneof`_ fields (mutually exclusive fields).
47+
For each oneof, at most one member field can be set at the same time.
48+
Setting any member of the oneof automatically clears all other
49+
members.
50+
51+
.. _oneof: https://proto-plus-python.readthedocs.io/en/stable/fields.html#oneofs-mutually-exclusive-fields
52+
4653
Attributes:
4754
optimization_objective_recall_value (float):
4855
Required when optimization_objective is
4956
"maximize-precision-at-recall". Must be between 0 and 1,
5057
inclusive.
58+
This field is a member of `oneof`_ ``additional_optimization_objective_config``.
5159
optimization_objective_precision_value (float):
5260
Required when optimization_objective is
5361
"maximize-recall-at-precision". Must be between 0 and 1,
5462
inclusive.
63+
This field is a member of `oneof`_ ``additional_optimization_objective_config``.
5564
prediction_type (str):
5665
The type of prediction the Model is to
5766
produce. "classification" - Predict one out of
@@ -149,23 +158,38 @@ class AutoMlTablesInputs(proto.Message):
149158
class Transformation(proto.Message):
150159
r"""
151160
161+
This message has `oneof`_ fields (mutually exclusive fields).
162+
For each oneof, at most one member field can be set at the same time.
163+
Setting any member of the oneof automatically clears all other
164+
members.
165+
166+
.. _oneof: https://proto-plus-python.readthedocs.io/en/stable/fields.html#oneofs-mutually-exclusive-fields
167+
152168
Attributes:
153169
auto (google.cloud.aiplatform.v1beta1.schema.trainingjob.definition_v1beta1.types.AutoMlTablesInputs.Transformation.AutoTransformation):
154170
171+
This field is a member of `oneof`_ ``transformation_detail``.
155172
numeric (google.cloud.aiplatform.v1beta1.schema.trainingjob.definition_v1beta1.types.AutoMlTablesInputs.Transformation.NumericTransformation):
156173
174+
This field is a member of `oneof`_ ``transformation_detail``.
157175
categorical (google.cloud.aiplatform.v1beta1.schema.trainingjob.definition_v1beta1.types.AutoMlTablesInputs.Transformation.CategoricalTransformation):
158176
177+
This field is a member of `oneof`_ ``transformation_detail``.
159178
timestamp (google.cloud.aiplatform.v1beta1.schema.trainingjob.definition_v1beta1.types.AutoMlTablesInputs.Transformation.TimestampTransformation):
160179
180+
This field is a member of `oneof`_ ``transformation_detail``.
161181
text (google.cloud.aiplatform.v1beta1.schema.trainingjob.definition_v1beta1.types.AutoMlTablesInputs.Transformation.TextTransformation):
162182
183+
This field is a member of `oneof`_ ``transformation_detail``.
163184
repeated_numeric (google.cloud.aiplatform.v1beta1.schema.trainingjob.definition_v1beta1.types.AutoMlTablesInputs.Transformation.NumericArrayTransformation):
164185
186+
This field is a member of `oneof`_ ``transformation_detail``.
165187
repeated_categorical (google.cloud.aiplatform.v1beta1.schema.trainingjob.definition_v1beta1.types.AutoMlTablesInputs.Transformation.CategoricalArrayTransformation):
166188
189+
This field is a member of `oneof`_ ``transformation_detail``.
167190
repeated_text (google.cloud.aiplatform.v1beta1.schema.trainingjob.definition_v1beta1.types.AutoMlTablesInputs.Transformation.TextArrayTransformation):
168191
192+
This field is a member of `oneof`_ ``transformation_detail``.
169193
"""
170194

171195
class AutoTransformation(proto.Message):

‎google/cloud/aiplatform/v1beta1/schema/trainingjob/definition_v1beta1/types/automl_time_series_forecasting.py‎

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -176,17 +176,29 @@ class AutoMlForecastingInputs(proto.Message):
176176
class Transformation(proto.Message):
177177
r"""
178178
179+
This message has `oneof`_ fields (mutually exclusive fields).
180+
For each oneof, at most one member field can be set at the same time.
181+
Setting any member of the oneof automatically clears all other
182+
members.
183+
184+
.. _oneof: https://proto-plus-python.readthedocs.io/en/stable/fields.html#oneofs-mutually-exclusive-fields
185+
179186
Attributes:
180187
auto (google.cloud.aiplatform.v1beta1.schema.trainingjob.definition_v1beta1.types.AutoMlForecastingInputs.Transformation.AutoTransformation):
181188
189+
This field is a member of `oneof`_ ``transformation_detail``.
182190
numeric (google.cloud.aiplatform.v1beta1.schema.trainingjob.definition_v1beta1.types.AutoMlForecastingInputs.Transformation.NumericTransformation):
183191
192+
This field is a member of `oneof`_ ``transformation_detail``.
184193
categorical (google.cloud.aiplatform.v1beta1.schema.trainingjob.definition_v1beta1.types.AutoMlForecastingInputs.Transformation.CategoricalTransformation):
185194
195+
This field is a member of `oneof`_ ``transformation_detail``.
186196
timestamp (google.cloud.aiplatform.v1beta1.schema.trainingjob.definition_v1beta1.types.AutoMlForecastingInputs.Transformation.TimestampTransformation):
187197
198+
This field is a member of `oneof`_ ``transformation_detail``.
188199
text (google.cloud.aiplatform.v1beta1.schema.trainingjob.definition_v1beta1.types.AutoMlForecastingInputs.Transformation.TextTransformation):
189200
201+
This field is a member of `oneof`_ ``transformation_detail``.
190202
"""
191203

192204
class AutoTransformation(proto.Message):

‎google/cloud/aiplatform_v1/services/dataset_service/async_client.py‎

Lines changed: 33 additions & 31 deletions
Original file line numberDiff line numberDiff line change
@@ -19,13 +19,15 @@
1919
from typing import Dict, Sequence, Tuple, Type, Union
2020
import pkg_resources
2121

22-
import google.api_core.client_options as ClientOptions # type: ignore
22+
from google.api_core.client_options import ClientOptions # type: ignore
2323
from google.api_core import exceptions as core_exceptions # type: ignore
2424
from google.api_core import gapic_v1 # type: ignore
2525
from google.api_core import retry as retries # type: ignore
2626
from google.auth import credentials as ga_credentials # type: ignore
2727
from google.oauth2 import service_account # type: ignore
2828

29+
OptionalRetry = Union[retries.Retry, object]
30+
2931
from google.api_core import operation as gac_operation # type: ignore
3032
from google.api_core import operation_async # type: ignore
3133
from google.cloud.aiplatform_v1.services.dataset_service import pagers
@@ -186,18 +188,18 @@ def __init__(
186188

187189
async def create_dataset(
188190
self,
189-
request: dataset_service.CreateDatasetRequest = None,
191+
request: Union[dataset_service.CreateDatasetRequest, dict] = None,
190192
*,
191193
parent: str = None,
192194
dataset: gca_dataset.Dataset = None,
193-
retry: retries.Retry = gapic_v1.method.DEFAULT,
195+
retry: OptionalRetry = gapic_v1.method.DEFAULT,
194196
timeout: float = None,
195197
metadata: Sequence[Tuple[str, str]] = (),
196198
) -> operation_async.AsyncOperation:
197199
r"""Creates a Dataset.
198200
199201
Args:
200-
request (:class:`google.cloud.aiplatform_v1.types.CreateDatasetRequest`):
202+
request (Union[google.cloud.aiplatform_v1.types.CreateDatasetRequest, dict]):
201203
The request object. Request message for
202204
[DatasetService.CreateDataset][google.cloud.aiplatform.v1.DatasetService.CreateDataset].
203205
parent (:class:`str`):
@@ -277,17 +279,17 @@ async def create_dataset(
277279

278280
async def get_dataset(
279281
self,
280-
request: dataset_service.GetDatasetRequest = None,
282+
request: Union[dataset_service.GetDatasetRequest, dict] = None,
281283
*,
282284
name: str = None,
283-
retry: retries.Retry = gapic_v1.method.DEFAULT,
285+
retry: OptionalRetry = gapic_v1.method.DEFAULT,
284286
timeout: float = None,
285287
metadata: Sequence[Tuple[str, str]] = (),
286288
) -> dataset.Dataset:
287289
r"""Gets a Dataset.
288290
289291
Args:
290-
request (:class:`google.cloud.aiplatform_v1.types.GetDatasetRequest`):
292+
request (Union[google.cloud.aiplatform_v1.types.GetDatasetRequest, dict]):
291293
The request object. Request message for
292294
[DatasetService.GetDataset][google.cloud.aiplatform.v1.DatasetService.GetDataset].
293295
name (:class:`str`):
@@ -348,18 +350,18 @@ async def get_dataset(
348350

349351
async def update_dataset(
350352
self,
351-
request: dataset_service.UpdateDatasetRequest = None,
353+
request: Union[dataset_service.UpdateDatasetRequest, dict] = None,
352354
*,
353355
dataset: gca_dataset.Dataset = None,
354356
update_mask: field_mask_pb2.FieldMask = None,
355-
retry: retries.Retry = gapic_v1.method.DEFAULT,
357+
retry: OptionalRetry = gapic_v1.method.DEFAULT,
356358
timeout: float = None,
357359
metadata: Sequence[Tuple[str, str]] = (),
358360
) -> gca_dataset.Dataset:
359361
r"""Updates a Dataset.
360362
361363
Args:
362-
request (:class:`google.cloud.aiplatform_v1.types.UpdateDatasetRequest`):
364+
request (Union[google.cloud.aiplatform_v1.types.UpdateDatasetRequest, dict]):
363365
The request object. Request message for
364366
[DatasetService.UpdateDataset][google.cloud.aiplatform.v1.DatasetService.UpdateDataset].
365367
dataset (:class:`google.cloud.aiplatform_v1.types.Dataset`):
@@ -437,17 +439,17 @@ async def update_dataset(
437439

438440
async def list_datasets(
439441
self,
440-
request: dataset_service.ListDatasetsRequest = None,
442+
request: Union[dataset_service.ListDatasetsRequest, dict] = None,
441443
*,
442444
parent: str = None,
443-
retry: retries.Retry = gapic_v1.method.DEFAULT,
445+
retry: OptionalRetry = gapic_v1.method.DEFAULT,
444446
timeout: float = None,
445447
metadata: Sequence[Tuple[str, str]] = (),
446448
) -> pagers.ListDatasetsAsyncPager:
447449
r"""Lists Datasets in a Location.
448450
449451
Args:
450-
request (:class:`google.cloud.aiplatform_v1.types.ListDatasetsRequest`):
452+
request (Union[google.cloud.aiplatform_v1.types.ListDatasetsRequest, dict]):
451453
The request object. Request message for
452454
[DatasetService.ListDatasets][google.cloud.aiplatform.v1.DatasetService.ListDatasets].
453455
parent (:class:`str`):
@@ -517,17 +519,17 @@ async def list_datasets(
517519

518520
async def delete_dataset(
519521
self,
520-
request: dataset_service.DeleteDatasetRequest = None,
522+
request: Union[dataset_service.DeleteDatasetRequest, dict] = None,
521523
*,
522524
name: str = None,
523-
retry: retries.Retry = gapic_v1.method.DEFAULT,
525+
retry: OptionalRetry = gapic_v1.method.DEFAULT,
524526
timeout: float = None,
525527
metadata: Sequence[Tuple[str, str]] = (),
526528
) -> operation_async.AsyncOperation:
527529
r"""Deletes a Dataset.
528530
529531
Args:
530-
request (:class:`google.cloud.aiplatform_v1.types.DeleteDatasetRequest`):
532+
request (Union[google.cloud.aiplatform_v1.types.DeleteDatasetRequest, dict]):
531533
The request object. Request message for
532534
[DatasetService.DeleteDataset][google.cloud.aiplatform.v1.DatasetService.DeleteDataset].
533535
name (:class:`str`):
@@ -610,18 +612,18 @@ async def delete_dataset(
610612

611613
async def import_data(
612614
self,
613-
request: dataset_service.ImportDataRequest = None,
615+
request: Union[dataset_service.ImportDataRequest, dict] = None,
614616
*,
615617
name: str = None,
616618
import_configs: Sequence[dataset.ImportDataConfig] = None,
617-
retry: retries.Retry = gapic_v1.method.DEFAULT,
619+
retry: OptionalRetry = gapic_v1.method.DEFAULT,
618620
timeout: float = None,
619621
metadata: Sequence[Tuple[str, str]] = (),
620622
) -> operation_async.AsyncOperation:
621623
r"""Imports data into a Dataset.
622624
623625
Args:
624-
request (:class:`google.cloud.aiplatform_v1.types.ImportDataRequest`):
626+
request (Union[google.cloud.aiplatform_v1.types.ImportDataRequest, dict]):
625627
The request object. Request message for
626628
[DatasetService.ImportData][google.cloud.aiplatform.v1.DatasetService.ImportData].
627629
name (:class:`str`):
@@ -704,18 +706,18 @@ async def import_data(
704706

705707
async def export_data(
706708
self,
707-
request: dataset_service.ExportDataRequest = None,
709+
request: Union[dataset_service.ExportDataRequest, dict] = None,
708710
*,
709711
name: str = None,
710712
export_config: dataset.ExportDataConfig = None,
711-
retry: retries.Retry = gapic_v1.method.DEFAULT,
713+
retry: OptionalRetry = gapic_v1.method.DEFAULT,
712714
timeout: float = None,
713715
metadata: Sequence[Tuple[str, str]] = (),
714716
) -> operation_async.AsyncOperation:
715717
r"""Exports data from a Dataset.
716718
717719
Args:
718-
request (:class:`google.cloud.aiplatform_v1.types.ExportDataRequest`):
720+
request (Union[google.cloud.aiplatform_v1.types.ExportDataRequest, dict]):
719721
The request object. Request message for
720722
[DatasetService.ExportData][google.cloud.aiplatform.v1.DatasetService.ExportData].
721723
name (:class:`str`):
@@ -797,17 +799,17 @@ async def export_data(
797799

798800
async def list_data_items(
799801
self,
800-
request: dataset_service.ListDataItemsRequest = None,
802+
request: Union[dataset_service.ListDataItemsRequest, dict] = None,
801803
*,
802804
parent: str = None,
803-
retry: retries.Retry = gapic_v1.method.DEFAULT,
805+
retry: OptionalRetry = gapic_v1.method.DEFAULT,
804806
timeout: float = None,
805807
metadata: Sequence[Tuple[str, str]] = (),
806808
) -> pagers.ListDataItemsAsyncPager:
807809
r"""Lists DataItems in a Dataset.
808810
809811
Args:
810-
request (:class:`google.cloud.aiplatform_v1.types.ListDataItemsRequest`):
812+
request (Union[google.cloud.aiplatform_v1.types.ListDataItemsRequest, dict]):
811813
The request object. Request message for
812814
[DatasetService.ListDataItems][google.cloud.aiplatform.v1.DatasetService.ListDataItems].
813815
parent (:class:`str`):
@@ -878,17 +880,17 @@ async def list_data_items(
878880

879881
async def get_annotation_spec(
880882
self,
881-
request: dataset_service.GetAnnotationSpecRequest = None,
883+
request: Union[dataset_service.GetAnnotationSpecRequest, dict] = None,
882884
*,
883885
name: str = None,
884-
retry: retries.Retry = gapic_v1.method.DEFAULT,
886+
retry: OptionalRetry = gapic_v1.method.DEFAULT,
885887
timeout: float = None,
886888
metadata: Sequence[Tuple[str, str]] = (),
887889
) -> annotation_spec.AnnotationSpec:
888890
r"""Gets an AnnotationSpec.
889891
890892
Args:
891-
request (:class:`google.cloud.aiplatform_v1.types.GetAnnotationSpecRequest`):
893+
request (Union[google.cloud.aiplatform_v1.types.GetAnnotationSpecRequest, dict]):
892894
The request object. Request message for
893895
[DatasetService.GetAnnotationSpec][google.cloud.aiplatform.v1.DatasetService.GetAnnotationSpec].
894896
name (:class:`str`):
@@ -950,17 +952,17 @@ async def get_annotation_spec(
950952

951953
async def list_annotations(
952954
self,
953-
request: dataset_service.ListAnnotationsRequest = None,
955+
request: Union[dataset_service.ListAnnotationsRequest, dict] = None,
954956
*,
955957
parent: str = None,
956-
retry: retries.Retry = gapic_v1.method.DEFAULT,
958+
retry: OptionalRetry = gapic_v1.method.DEFAULT,
957959
timeout: float = None,
958960
metadata: Sequence[Tuple[str, str]] = (),
959961
) -> pagers.ListAnnotationsAsyncPager:
960962
r"""Lists Annotations belongs to a dataitem
961963
962964
Args:
963-
request (:class:`google.cloud.aiplatform_v1.types.ListAnnotationsRequest`):
965+
request (Union[google.cloud.aiplatform_v1.types.ListAnnotationsRequest, dict]):
964966
The request object. Request message for
965967
[DatasetService.ListAnnotations][google.cloud.aiplatform.v1.DatasetService.ListAnnotations].
966968
parent (:class:`str`):

0 commit comments

Comments
 (0)