Skip to content

Commit 593b54f

Browse files
yinghsienwucopybara-github
authored andcommitted
chore: add PersistentResourceServiceClient to VertexAIServiceClientWithOverride
PiperOrigin-RevId: 556965189
1 parent 65e263c commit 593b54f

File tree

2 files changed

+17
-0
lines changed

2 files changed

+17
-0
lines changed

google/cloud/aiplatform/compat/services/__init__.py

+4
Original file line numberDiff line numberDiff line change
@@ -54,6 +54,9 @@
5454
from google.cloud.aiplatform_v1beta1.services.pipeline_service import (
5555
client as pipeline_service_client_v1beta1,
5656
)
57+
from google.cloud.aiplatform_v1beta1.services.persistent_resource_service import (
58+
client as persistent_resource_service_client_v1beta1,
59+
)
5760
from google.cloud.aiplatform_v1beta1.services.prediction_service import (
5861
client as prediction_service_client_v1beta1,
5962
)
@@ -145,6 +148,7 @@
145148
match_service_client_v1beta1,
146149
model_garden_service_client_v1beta1,
147150
model_service_client_v1beta1,
151+
persistent_resource_service_client_v1beta1,
148152
pipeline_service_client_v1beta1,
149153
prediction_service_client_v1beta1,
150154
schedule_service_client_v1beta1,

google/cloud/aiplatform/utils/__init__.py

+13
Original file line numberDiff line numberDiff line change
@@ -53,6 +53,7 @@
5353
tensorboard_service_client_v1beta1,
5454
vizier_service_client_v1beta1,
5555
model_garden_service_client_v1beta1,
56+
persistent_resource_service_client_v1beta1,
5657
)
5758
from google.cloud.aiplatform.compat.services import (
5859
dataset_service_client_v1,
@@ -654,6 +655,17 @@ class ModelGardenClientWithOverride(ClientWithOverride):
654655
)
655656

656657

658+
class PersistentResourceClientWithOverride(ClientWithOverride):
659+
_is_temporary = True
660+
_default_version = compat.V1BETA1
661+
_version_map = (
662+
(
663+
compat.V1BETA1,
664+
persistent_resource_service_client_v1beta1.PersistentResourceServiceClient,
665+
),
666+
)
667+
668+
657669
VertexAiServiceClientWithOverride = TypeVar(
658670
"VertexAiServiceClientWithOverride",
659671
DatasetClientWithOverride,
@@ -670,6 +682,7 @@ class ModelGardenClientWithOverride(ClientWithOverride):
670682
TensorboardClientWithOverride,
671683
VizierClientWithOverride,
672684
ModelGardenClientWithOverride,
685+
PersistentResourceClientWithOverride,
673686
)
674687

675688

0 commit comments

Comments
 (0)