Skip to content

Commit f3a3d03

Browse files
gcf-owl-bot[bot]gcf-owl-bot[bot]
authored
feat: Removes AcceleratorType.TPU_V2 and TPU_V3 constants (#543)
PiperOrigin-RevId: 384714314 Source-Link: googleapis/googleapis@cc51e48 Source-Link: googleapis/googleapis-gen@f87b91c feat: Adds AcceleratorType.NVIDIA_TESLA_A100 constant feat: Adds BigQuery output table field to batch prediction job output config feat: Adds JobState.JOB_STATE_EXPIRED constant feat: Adds AutoscalingMetricSpec message feat: Adds PipelineService methods for Create, Get, List, Delete, Cancel feat: Adds fields to Study message
1 parent 41093b7 commit f3a3d03

28 files changed

+4052
-108
lines changed

‎google/cloud/aiplatform_v1/__init__.py‎

Lines changed: 30 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -34,8 +34,10 @@
3434
from .types.accelerator_type import AcceleratorType
3535
from .types.annotation import Annotation
3636
from .types.annotation_spec import AnnotationSpec
37+
from .types.artifact import Artifact
3738
from .types.batch_prediction_job import BatchPredictionJob
3839
from .types.completion_stats import CompletionStats
40+
from .types.context import Context
3941
from .types.custom_job import ContainerSpec
4042
from .types.custom_job import CustomJob
4143
from .types.custom_job import CustomJobSpec
@@ -86,6 +88,7 @@
8688
from .types.endpoint_service import UndeployModelResponse
8789
from .types.endpoint_service import UpdateEndpointRequest
8890
from .types.env_var import EnvVar
91+
from .types.execution import Execution
8992
from .types.hyperparameter_tuning_job import HyperparameterTuningJob
9093
from .types.io import BigQueryDestination
9194
from .types.io import BigQuerySource
@@ -118,6 +121,7 @@
118121
from .types.job_service import ListHyperparameterTuningJobsResponse
119122
from .types.job_state import JobState
120123
from .types.machine_resources import AutomaticResources
124+
from .types.machine_resources import AutoscalingMetricSpec
121125
from .types.machine_resources import BatchDedicatedResources
122126
from .types.machine_resources import DedicatedResources
123127
from .types.machine_resources import DiskSpec
@@ -157,10 +161,20 @@
157161
from .types.model_service import UploadModelResponse
158162
from .types.operation import DeleteOperationMetadata
159163
from .types.operation import GenericOperationMetadata
164+
from .types.pipeline_job import PipelineJob
165+
from .types.pipeline_job import PipelineJobDetail
166+
from .types.pipeline_job import PipelineTaskDetail
167+
from .types.pipeline_job import PipelineTaskExecutorDetail
168+
from .types.pipeline_service import CancelPipelineJobRequest
160169
from .types.pipeline_service import CancelTrainingPipelineRequest
170+
from .types.pipeline_service import CreatePipelineJobRequest
161171
from .types.pipeline_service import CreateTrainingPipelineRequest
172+
from .types.pipeline_service import DeletePipelineJobRequest
162173
from .types.pipeline_service import DeleteTrainingPipelineRequest
174+
from .types.pipeline_service import GetPipelineJobRequest
163175
from .types.pipeline_service import GetTrainingPipelineRequest
176+
from .types.pipeline_service import ListPipelineJobsRequest
177+
from .types.pipeline_service import ListPipelineJobsResponse
164178
from .types.pipeline_service import ListTrainingPipelinesRequest
165179
from .types.pipeline_service import ListTrainingPipelinesResponse
166180
from .types.pipeline_state import PipelineState
@@ -185,6 +199,7 @@
185199
from .types.training_pipeline import TimestampSplit
186200
from .types.training_pipeline import TrainingPipeline
187201
from .types.user_action_reference import UserActionReference
202+
from .types.value import Value
188203

189204
__all__ = (
190205
"DatasetServiceAsyncClient",
@@ -199,7 +214,9 @@
199214
"ActiveLearningConfig",
200215
"Annotation",
201216
"AnnotationSpec",
217+
"Artifact",
202218
"AutomaticResources",
219+
"AutoscalingMetricSpec",
203220
"BatchDedicatedResources",
204221
"BatchMigrateResourcesOperationMetadata",
205222
"BatchMigrateResourcesRequest",
@@ -211,10 +228,12 @@
211228
"CancelCustomJobRequest",
212229
"CancelDataLabelingJobRequest",
213230
"CancelHyperparameterTuningJobRequest",
231+
"CancelPipelineJobRequest",
214232
"CancelTrainingPipelineRequest",
215233
"CompletionStats",
216234
"ContainerRegistryDestination",
217235
"ContainerSpec",
236+
"Context",
218237
"CreateBatchPredictionJobRequest",
219238
"CreateCustomJobRequest",
220239
"CreateDataLabelingJobRequest",
@@ -223,6 +242,7 @@
223242
"CreateEndpointOperationMetadata",
224243
"CreateEndpointRequest",
225244
"CreateHyperparameterTuningJobRequest",
245+
"CreatePipelineJobRequest",
226246
"CreateSpecialistPoolOperationMetadata",
227247
"CreateSpecialistPoolRequest",
228248
"CreateTrainingPipelineRequest",
@@ -241,6 +261,7 @@
241261
"DeleteHyperparameterTuningJobRequest",
242262
"DeleteModelRequest",
243263
"DeleteOperationMetadata",
264+
"DeletePipelineJobRequest",
244265
"DeleteSpecialistPoolRequest",
245266
"DeleteTrainingPipelineRequest",
246267
"DeployModelOperationMetadata",
@@ -253,6 +274,7 @@
253274
"Endpoint",
254275
"EndpointServiceClient",
255276
"EnvVar",
277+
"Execution",
256278
"ExportDataConfig",
257279
"ExportDataOperationMetadata",
258280
"ExportDataRequest",
@@ -275,6 +297,7 @@
275297
"GetModelEvaluationRequest",
276298
"GetModelEvaluationSliceRequest",
277299
"GetModelRequest",
300+
"GetPipelineJobRequest",
278301
"GetSpecialistPoolRequest",
279302
"GetTrainingPipelineRequest",
280303
"HyperparameterTuningJob",
@@ -307,6 +330,8 @@
307330
"ListModelEvaluationsResponse",
308331
"ListModelsRequest",
309332
"ListModelsResponse",
333+
"ListPipelineJobsRequest",
334+
"ListPipelineJobsResponse",
310335
"ListSpecialistPoolsRequest",
311336
"ListSpecialistPoolsResponse",
312337
"ListTrainingPipelinesRequest",
@@ -323,8 +348,12 @@
323348
"ModelEvaluation",
324349
"ModelEvaluationSlice",
325350
"ModelServiceClient",
351+
"PipelineJob",
352+
"PipelineJobDetail",
326353
"PipelineServiceClient",
327354
"PipelineState",
355+
"PipelineTaskDetail",
356+
"PipelineTaskExecutorDetail",
328357
"Port",
329358
"PredefinedSplit",
330359
"PredictRequest",
@@ -356,5 +385,6 @@
356385
"UploadModelRequest",
357386
"UploadModelResponse",
358387
"UserActionReference",
388+
"Value",
359389
"WorkerPoolSpec",
360390
)

‎google/cloud/aiplatform_v1/gapic_metadata.json‎

Lines changed: 50 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -570,26 +570,51 @@
570570
"grpc": {
571571
"libraryClient": "PipelineServiceClient",
572572
"rpcs": {
573+
"CancelPipelineJob": {
574+
"methods": [
575+
"cancel_pipeline_job"
576+
]
577+
},
573578
"CancelTrainingPipeline": {
574579
"methods": [
575580
"cancel_training_pipeline"
576581
]
577582
},
583+
"CreatePipelineJob": {
584+
"methods": [
585+
"create_pipeline_job"
586+
]
587+
},
578588
"CreateTrainingPipeline": {
579589
"methods": [
580590
"create_training_pipeline"
581591
]
582592
},
593+
"DeletePipelineJob": {
594+
"methods": [
595+
"delete_pipeline_job"
596+
]
597+
},
583598
"DeleteTrainingPipeline": {
584599
"methods": [
585600
"delete_training_pipeline"
586601
]
587602
},
603+
"GetPipelineJob": {
604+
"methods": [
605+
"get_pipeline_job"
606+
]
607+
},
588608
"GetTrainingPipeline": {
589609
"methods": [
590610
"get_training_pipeline"
591611
]
592612
},
613+
"ListPipelineJobs": {
614+
"methods": [
615+
"list_pipeline_jobs"
616+
]
617+
},
593618
"ListTrainingPipelines": {
594619
"methods": [
595620
"list_training_pipelines"
@@ -600,26 +625,51 @@
600625
"grpc-async": {
601626
"libraryClient": "PipelineServiceAsyncClient",
602627
"rpcs": {
628+
"CancelPipelineJob": {
629+
"methods": [
630+
"cancel_pipeline_job"
631+
]
632+
},
603633
"CancelTrainingPipeline": {
604634
"methods": [
605635
"cancel_training_pipeline"
606636
]
607637
},
638+
"CreatePipelineJob": {
639+
"methods": [
640+
"create_pipeline_job"
641+
]
642+
},
608643
"CreateTrainingPipeline": {
609644
"methods": [
610645
"create_training_pipeline"
611646
]
612647
},
648+
"DeletePipelineJob": {
649+
"methods": [
650+
"delete_pipeline_job"
651+
]
652+
},
613653
"DeleteTrainingPipeline": {
614654
"methods": [
615655
"delete_training_pipeline"
616656
]
617657
},
658+
"GetPipelineJob": {
659+
"methods": [
660+
"get_pipeline_job"
661+
]
662+
},
618663
"GetTrainingPipeline": {
619664
"methods": [
620665
"get_training_pipeline"
621666
]
622667
},
668+
"ListPipelineJobs": {
669+
"methods": [
670+
"list_pipeline_jobs"
671+
]
672+
},
623673
"ListTrainingPipelines": {
624674
"methods": [
625675
"list_training_pipelines"

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

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -87,6 +87,8 @@ class JobServiceAsyncClient:
8787
)
8888
model_path = staticmethod(JobServiceClient.model_path)
8989
parse_model_path = staticmethod(JobServiceClient.parse_model_path)
90+
network_path = staticmethod(JobServiceClient.network_path)
91+
parse_network_path = staticmethod(JobServiceClient.parse_network_path)
9092
trial_path = staticmethod(JobServiceClient.trial_path)
9193
parse_trial_path = staticmethod(JobServiceClient.parse_trial_path)
9294
common_billing_account_path = staticmethod(

‎google/cloud/aiplatform_v1/services/job_service/client.py‎

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -285,6 +285,21 @@ def parse_model_path(path: str) -> Dict[str, str]:
285285
)
286286
return m.groupdict() if m else {}
287287

288+
@staticmethod
289+
def network_path(project: str, network: str,) -> str:
290+
"""Returns a fully-qualified network string."""
291+
return "projects/{project}/global/networks/{network}".format(
292+
project=project, network=network,
293+
)
294+
295+
@staticmethod
296+
def parse_network_path(path: str) -> Dict[str, str]:
297+
"""Parses a network path into its component segments."""
298+
m = re.match(
299+
r"^projects/(?P<project>.+?)/global/networks/(?P<network>.+?)$", path
300+
)
301+
return m.groupdict() if m else {}
302+
288303
@staticmethod
289304
def trial_path(project: str, location: str, study: str, trial: str,) -> str:
290305
"""Returns a fully-qualified trial string."""

0 commit comments

Comments
 (0)