feat: add reserved_ip_ranges to CustomJobSpec in aiplatform v1 custom…

…_job.proto (#1165) - [ ] Regenerate this pull request now. PiperOrigin-RevId: 443508623 Source-Link: googleapis/googleapis@fd6935f Source-Link: googleapis/googleapis-gen@4f659e3 Copy-Tag: eyJwIjoiLmdpdGh1Yi8uT3dsQm90LnlhbWwiLCJoIjoiNGY2NTllM2I4YWNlMmU4NmMwMjdjNGZlY2E5ZGVjZDZlZTU4YzY0YyJ9 PiperOrigin-RevId: 442088600 Source-Link: googleapis/googleapis@c56ae2a Source-Link: googleapis/googleapis-gen@12d6b9d Copy-Tag: eyJwIjoiLmdpdGh1Yi8uT3dsQm90LnlhbWwiLCJoIjoiMTJkNmI5ZGZlNzBjNzY5ZGEzZTAzMDJjZmQzNWQxN2Y2MzNlM2E1OSJ9 feat: add nfs_mounts to WorkPoolSpec in aiplatform v1 custom_job.proto feat: add JOB_STATE_UPDATING to JobState in aiplatform v1 job_state.proto feat: add MfsMount in aiplatform v1 machine_resources.proto feat: add ConvexAutomatedStoppingSpec to StudySpec in aiplatform v1 study.proto PiperOrigin-RevId: 441616654 Source-Link: googleapis/googleapis@a476b9f Source-Link: googleapis/googleapis-gen@ed936e9 Copy-Tag: eyJwIjoiLmdpdGh1Yi8uT3dsQm90LnlhbWwiLCJoIjoiZWQ5MzZlOTA1MjEyZTUwZmViODYyOTEyMWRmN2JlOTQzMGRiZDhhYSJ9 feat: rename Similarity to Examples, and similarity to examples in ExplanationParameters in aiplatform v1beta1 explanation.proto feat: add reserved_ip_ranges to CustomJobSpec in aiplatform v1beta1 custom_job.proto feat: add nfs_mounts to WorkPoolSpec in aiplatform v1beta1 custom_job.proto feat: add PredictRequestResponseLoggingConfig to aiplatform v1beta1 endpoint.proto feat: add model_version_id to DeployedModel in aiplatform v1beta1 endpoint.proto feat: add JOB_STATE_UPDATING to JobState in aiplatform v1beta1 job_state.proto feat: add MfsMount in aiplatform v1beta1 machine_resources.proto feat: add version_id to Model in aiplatform v1beta1 model.proto feat: add LatestMonitoringPipelineMetadata to ModelDeploymentMonitoringJob in aiplatform v1beta1 model_deployment_monitoring_job.proto feat: add ListModelVersion, DeleteModelVersion, and MergeVersionAliases rpcs to aiplatform v1beta1 model_service.proto feat: add model_version_id to UploadModelRequest and UploadModelResponse in aiplatform v1beta1 model_service.proto feat: add model_version_id to PredictResponse in aiplatform v1beta1 prediction_service.proto feat: add ConvexAutomatedStoppingSpec to StudySpec in aiplatform v1beta1 study.proto feat: add model_id and parent_model to TrainingPipeline in aiplatform v1beta1 training_pipeline.proto PiperOrigin-RevId: 441615103 Source-Link: googleapis/googleapis@2ce1c56 Source-Link: googleapis/googleapis-gen@54ad530 Copy-Tag: eyJwIjoiLmdpdGh1Yi8uT3dsQm90LnlhbWwiLCJoIjoiNTRhZDUzMDI2ODE2OTdkNGU0NWJlNjkwNjkzMjJkOTMyOGI1ODU4MCJ9 chore: use gapic-generator-python 0.65.1 PiperOrigin-RevId: 441524537 Source-Link: googleapis/googleapis@2a27391 Source-Link: googleapis/googleapis-gen@ab6756a Copy-Tag: eyJwIjoiLmdpdGh1Yi8uT3dsQm90LnlhbWwiLCJoIjoiYWI2NzU2YTQ4Yzg5YjViY2I5ZmI3MzQ0M2NiOGU1NWQ1NzRmNDY0MyJ9 docs: fix type in docstring for map fields
googleapis · Apr 25, 2022 · 847ad78 · 847ad78
1 parent 3224ae3
commit 847ad78
Show file tree

Hide file tree

Showing 242 changed files with 36,400 additions and 3,035 deletions.
diff --git a/google/cloud/aiplatform/v1/schema/predict/instance_v1/types/__init__.py b/google/cloud/aiplatform/v1/schema/predict/instance_v1/types/__init__.py
@@ -13,15 +13,33 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
-from .image_classification import ImageClassificationPredictionInstance
-from .image_object_detection import ImageObjectDetectionPredictionInstance
-from .image_segmentation import ImageSegmentationPredictionInstance
-from .text_classification import TextClassificationPredictionInstance
-from .text_extraction import TextExtractionPredictionInstance
-from .text_sentiment import TextSentimentPredictionInstance
-from .video_action_recognition import VideoActionRecognitionPredictionInstance
-from .video_classification import VideoClassificationPredictionInstance
-from .video_object_tracking import VideoObjectTrackingPredictionInstance
+from .image_classification import (
+    ImageClassificationPredictionInstance,
+)
+from .image_object_detection import (
+    ImageObjectDetectionPredictionInstance,
+)
+from .image_segmentation import (
+    ImageSegmentationPredictionInstance,
+)
+from .text_classification import (
+    TextClassificationPredictionInstance,
+)
+from .text_extraction import (
+    TextExtractionPredictionInstance,
+)
+from .text_sentiment import (
+    TextSentimentPredictionInstance,
+)
+from .video_action_recognition import (
+    VideoActionRecognitionPredictionInstance,
+)
+from .video_classification import (
+    VideoClassificationPredictionInstance,
+)
+from .video_object_tracking import (
+    VideoObjectTrackingPredictionInstance,
+)
 
 __all__ = (
     "ImageClassificationPredictionInstance",

diff --git a/google/cloud/aiplatform/v1/schema/predict/params_v1/types/__init__.py b/google/cloud/aiplatform/v1/schema/predict/params_v1/types/__init__.py
@@ -13,12 +13,24 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
-from .image_classification import ImageClassificationPredictionParams
-from .image_object_detection import ImageObjectDetectionPredictionParams
-from .image_segmentation import ImageSegmentationPredictionParams
-from .video_action_recognition import VideoActionRecognitionPredictionParams
-from .video_classification import VideoClassificationPredictionParams
-from .video_object_tracking import VideoObjectTrackingPredictionParams
+from .image_classification import (
+    ImageClassificationPredictionParams,
+)
+from .image_object_detection import (
+    ImageObjectDetectionPredictionParams,
+)
+from .image_segmentation import (
+    ImageSegmentationPredictionParams,
+)
+from .video_action_recognition import (
+    VideoActionRecognitionPredictionParams,
+)
+from .video_classification import (
+    VideoClassificationPredictionParams,
+)
+from .video_object_tracking import (
+    VideoObjectTrackingPredictionParams,
+)
 
 __all__ = (
     "ImageClassificationPredictionParams",

diff --git a/google/cloud/aiplatform/v1/schema/predict/prediction_v1/types/__init__.py b/google/cloud/aiplatform/v1/schema/predict/prediction_v1/types/__init__.py
@@ -13,16 +13,36 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
-from .classification import ClassificationPredictionResult
-from .image_object_detection import ImageObjectDetectionPredictionResult
-from .image_segmentation import ImageSegmentationPredictionResult
-from .tabular_classification import TabularClassificationPredictionResult
-from .tabular_regression import TabularRegressionPredictionResult
-from .text_extraction import TextExtractionPredictionResult
-from .text_sentiment import TextSentimentPredictionResult
-from .video_action_recognition import VideoActionRecognitionPredictionResult
-from .video_classification import VideoClassificationPredictionResult
-from .video_object_tracking import VideoObjectTrackingPredictionResult
+from .classification import (
+    ClassificationPredictionResult,
+)
+from .image_object_detection import (
+    ImageObjectDetectionPredictionResult,
+)
+from .image_segmentation import (
+    ImageSegmentationPredictionResult,
+)
+from .tabular_classification import (
+    TabularClassificationPredictionResult,
+)
+from .tabular_regression import (
+    TabularRegressionPredictionResult,
+)
+from .text_extraction import (
+    TextExtractionPredictionResult,
+)
+from .text_sentiment import (
+    TextSentimentPredictionResult,
+)
+from .video_action_recognition import (
+    VideoActionRecognitionPredictionResult,
+)
+from .video_classification import (
+    VideoClassificationPredictionResult,
+)
+from .video_object_tracking import (
+    VideoObjectTrackingPredictionResult,
+)
 
 __all__ = (
     "ClassificationPredictionResult",

diff --git a/google/cloud/aiplatform/v1/schema/trainingjob/definition_v1/types/__init__.py b/google/cloud/aiplatform/v1/schema/trainingjob/definition_v1/types/__init__.py
@@ -57,7 +57,9 @@
     AutoMlVideoObjectTracking,
     AutoMlVideoObjectTrackingInputs,
 )
-from .export_evaluated_data_items_config import ExportEvaluatedDataItemsConfig
+from .export_evaluated_data_items_config import (
+    ExportEvaluatedDataItemsConfig,
+)
 
 __all__ = (
     "AutoMlImageClassification",

diff --git a/google/cloud/aiplatform/v1beta1/schema/predict/instance_v1beta1/types/__init__.py b/google/cloud/aiplatform/v1beta1/schema/predict/instance_v1beta1/types/__init__.py
@@ -13,15 +13,33 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
-from .image_classification import ImageClassificationPredictionInstance
-from .image_object_detection import ImageObjectDetectionPredictionInstance
-from .image_segmentation import ImageSegmentationPredictionInstance
-from .text_classification import TextClassificationPredictionInstance
-from .text_extraction import TextExtractionPredictionInstance
-from .text_sentiment import TextSentimentPredictionInstance
-from .video_action_recognition import VideoActionRecognitionPredictionInstance
-from .video_classification import VideoClassificationPredictionInstance
-from .video_object_tracking import VideoObjectTrackingPredictionInstance
+from .image_classification import (
+    ImageClassificationPredictionInstance,
+)
+from .image_object_detection import (
+    ImageObjectDetectionPredictionInstance,
+)
+from .image_segmentation import (
+    ImageSegmentationPredictionInstance,
+)
+from .text_classification import (
+    TextClassificationPredictionInstance,
+)
+from .text_extraction import (
+    TextExtractionPredictionInstance,
+)
+from .text_sentiment import (
+    TextSentimentPredictionInstance,
+)
+from .video_action_recognition import (
+    VideoActionRecognitionPredictionInstance,
+)
+from .video_classification import (
+    VideoClassificationPredictionInstance,
+)
+from .video_object_tracking import (
+    VideoObjectTrackingPredictionInstance,
+)
 
 __all__ = (
     "ImageClassificationPredictionInstance",

diff --git a/google/cloud/aiplatform/v1beta1/schema/predict/params_v1beta1/types/__init__.py b/google/cloud/aiplatform/v1beta1/schema/predict/params_v1beta1/types/__init__.py
@@ -13,12 +13,24 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
-from .image_classification import ImageClassificationPredictionParams
-from .image_object_detection import ImageObjectDetectionPredictionParams
-from .image_segmentation import ImageSegmentationPredictionParams
-from .video_action_recognition import VideoActionRecognitionPredictionParams
-from .video_classification import VideoClassificationPredictionParams
-from .video_object_tracking import VideoObjectTrackingPredictionParams
+from .image_classification import (
+    ImageClassificationPredictionParams,
+)
+from .image_object_detection import (
+    ImageObjectDetectionPredictionParams,
+)
+from .image_segmentation import (
+    ImageSegmentationPredictionParams,
+)
+from .video_action_recognition import (
+    VideoActionRecognitionPredictionParams,
+)
+from .video_classification import (
+    VideoClassificationPredictionParams,
+)
+from .video_object_tracking import (
+    VideoObjectTrackingPredictionParams,
+)
 
 __all__ = (
     "ImageClassificationPredictionParams",

diff --git a/google/cloud/aiplatform/v1beta1/schema/predict/prediction_v1beta1/types/__init__.py b/google/cloud/aiplatform/v1beta1/schema/predict/prediction_v1beta1/types/__init__.py
@@ -13,17 +13,39 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
-from .classification import ClassificationPredictionResult
-from .image_object_detection import ImageObjectDetectionPredictionResult
-from .image_segmentation import ImageSegmentationPredictionResult
-from .tabular_classification import TabularClassificationPredictionResult
-from .tabular_regression import TabularRegressionPredictionResult
-from .text_extraction import TextExtractionPredictionResult
-from .text_sentiment import TextSentimentPredictionResult
-from .time_series_forecasting import TimeSeriesForecastingPredictionResult
-from .video_action_recognition import VideoActionRecognitionPredictionResult
-from .video_classification import VideoClassificationPredictionResult
-from .video_object_tracking import VideoObjectTrackingPredictionResult
+from .classification import (
+    ClassificationPredictionResult,
+)
+from .image_object_detection import (
+    ImageObjectDetectionPredictionResult,
+)
+from .image_segmentation import (
+    ImageSegmentationPredictionResult,
+)
+from .tabular_classification import (
+    TabularClassificationPredictionResult,
+)
+from .tabular_regression import (
+    TabularRegressionPredictionResult,
+)
+from .text_extraction import (
+    TextExtractionPredictionResult,
+)
+from .text_sentiment import (
+    TextSentimentPredictionResult,
+)
+from .time_series_forecasting import (
+    TimeSeriesForecastingPredictionResult,
+)
+from .video_action_recognition import (
+    VideoActionRecognitionPredictionResult,
+)
+from .video_classification import (
+    VideoClassificationPredictionResult,
+)
+from .video_object_tracking import (
+    VideoObjectTrackingPredictionResult,
+)
 
 __all__ = (
     "ClassificationPredictionResult",

diff --git a/google/cloud/aiplatform/v1beta1/schema/trainingjob/definition_v1beta1/types/__init__.py b/google/cloud/aiplatform/v1beta1/schema/trainingjob/definition_v1beta1/types/__init__.py
@@ -62,7 +62,9 @@
     AutoMlVideoObjectTracking,
     AutoMlVideoObjectTrackingInputs,
 )
-from .export_evaluated_data_items_config import ExportEvaluatedDataItemsConfig
+from .export_evaluated_data_items_config import (
+    ExportEvaluatedDataItemsConfig,
+)
 
 __all__ = (
     "AutoMlImageClassification",

diff --git a/google/cloud/aiplatform_v1/__init__.py b/google/cloud/aiplatform_v1/__init__.py
@@ -258,6 +258,7 @@
 from .types.machine_resources import DedicatedResources
 from .types.machine_resources import DiskSpec
 from .types.machine_resources import MachineSpec
+from .types.machine_resources import NfsMount
 from .types.machine_resources import ResourcesConsumed
 from .types.manual_batch_tuning_parameters import ManualBatchTuningParameters
 from .types.metadata_schema import MetadataSchema
@@ -826,6 +827,7 @@
     "MutateDeployedIndexRequest",
     "MutateDeployedIndexResponse",
     "NearestNeighborSearchOperationMetadata",
+    "NfsMount",
     "PauseModelDeploymentMonitoringJobRequest",
     "PipelineJob",
     "PipelineJobDetail",

diff --git a/google/cloud/aiplatform_v1/services/dataset_service/async_client.py b/google/cloud/aiplatform_v1/services/dataset_service/async_client.py
@@ -16,7 +16,7 @@
 from collections import OrderedDict
 import functools
 import re
-from typing import Dict, Optional, Sequence, Tuple, Type, Union
+from typing import Dict, Mapping, Optional, Sequence, Tuple, Type, Union
 import pkg_resources
 
 from google.api_core.client_options import ClientOptions

diff --git a/google/cloud/aiplatform_v1/services/dataset_service/client.py b/google/cloud/aiplatform_v1/services/dataset_service/client.py
@@ -16,7 +16,7 @@
 from collections import OrderedDict
 import os
 import re
-from typing import Dict, Optional, Sequence, Tuple, Type, Union
+from typing import Dict, Mapping, Optional, Sequence, Tuple, Type, Union
 import pkg_resources
 
 from google.api_core import client_options as client_options_lib

diff --git a/google/cloud/aiplatform_v1/services/dataset_service/transports/base.py b/google/cloud/aiplatform_v1/services/dataset_service/transports/base.py
@@ -85,6 +85,7 @@ def __init__(
             always_use_jwt_access (Optional[bool]): Whether self signed JWT should
                 be used for service account credentials.
         """
+
         # Save the hostname. Default to port 443 (HTTPS) if none is specified.
         if ":" not in host:
             host += ":443"
@@ -292,5 +293,9 @@ def list_annotations(
     ]:
         raise NotImplementedError()
 
+    @property
+    def kind(self) -> str:
+        raise NotImplementedError()
+
 
 __all__ = ("DatasetServiceTransport",)
diff --git a/google/cloud/aiplatform_v1/services/dataset_service/transports/grpc.py b/google/cloud/aiplatform_v1/services/dataset_service/transports/grpc.py
@@ -520,5 +520,9 @@ def list_annotations(
     def close(self):
         self.grpc_channel.close()
 
+    @property
+    def kind(self) -> str:
+        return "grpc"
+
 
 __all__ = ("DatasetServiceGrpcTransport",)
diff --git a/google/cloud/aiplatform_v1/services/endpoint_service/async_client.py b/google/cloud/aiplatform_v1/services/endpoint_service/async_client.py
@@ -16,7 +16,7 @@
 from collections import OrderedDict
 import functools
 import re
-from typing import Dict, Optional, Sequence, Tuple, Type, Union
+from typing import Dict, Mapping, Optional, Sequence, Tuple, Type, Union
 import pkg_resources
 
 from google.api_core.client_options import ClientOptions
@@ -804,17 +804,14 @@ async def deploy_model(
         *,
         endpoint: str = None,
         deployed_model: gca_endpoint.DeployedModel = None,
-        traffic_split: Sequence[
-            endpoint_service.DeployModelRequest.TrafficSplitEntry
-        ] = None,
+        traffic_split: Mapping[str, int] = None,
         retry: OptionalRetry = gapic_v1.method.DEFAULT,
         timeout: float = None,
         metadata: Sequence[Tuple[str, str]] = (),
     ) -> operation_async.AsyncOperation:
         r"""Deploys a Model into this Endpoint, creating a
         DeployedModel within it.
 
-
         .. code-block:: python
 
             from google.cloud import aiplatform_v1
@@ -866,7 +863,7 @@ def sample_deploy_model():
                 This corresponds to the ``deployed_model`` field
                 on the ``request`` instance; if ``request`` is provided, this
                 should not be set.
-            traffic_split (:class:`Sequence[google.cloud.aiplatform_v1.types.DeployModelRequest.TrafficSplitEntry]`):
+            traffic_split (:class:`Mapping[str, int]`):
                 A map from a DeployedModel's ID to the percentage of
                 this Endpoint's traffic that should be forwarded to that
                 DeployedModel.
@@ -963,9 +960,7 @@ async def undeploy_model(
         *,
         endpoint: str = None,
         deployed_model_id: str = None,
-        traffic_split: Sequence[
-            endpoint_service.UndeployModelRequest.TrafficSplitEntry
-        ] = None,
+        traffic_split: Mapping[str, int] = None,
         retry: OptionalRetry = gapic_v1.method.DEFAULT,
         timeout: float = None,
         metadata: Sequence[Tuple[str, str]] = (),
@@ -974,7 +969,6 @@ async def undeploy_model(
         DeployedModel from it, and freeing all resources it's
         using.
 
-
         .. code-block:: python
 
             from google.cloud import aiplatform_v1
@@ -1018,7 +1012,7 @@ def sample_undeploy_model():
                 This corresponds to the ``deployed_model_id`` field
                 on the ``request`` instance; if ``request`` is provided, this
                 should not be set.
-            traffic_split (:class:`Sequence[google.cloud.aiplatform_v1.types.UndeployModelRequest.TrafficSplitEntry]`):
+            traffic_split (:class:`Mapping[str, int]`):
                 If this field is provided, then the Endpoint's
                 [traffic_split][google.cloud.aiplatform.v1.Endpoint.traffic_split]
                 will be overwritten with it. If last DeployedModel is