Skip to content

Commit 1474d98

Browse files
vertex-sdk-botcopybara-github
authored andcommitted
feat: Migrate DeploymentResourcePool and associated functionality to V1 namespace for GA launch of model co-hosting.
PiperOrigin-RevId: 649164615
1 parent b63f960 commit 1474d98

File tree

9 files changed

+785
-109
lines changed

9 files changed

+785
-109
lines changed

google/cloud/aiplatform/compat/__init__.py

+8
Original file line numberDiff line numberDiff line change
@@ -80,6 +80,9 @@
8080
types.dataset_service = types.dataset_service_v1beta1
8181
types.deployed_model_ref = types.deployed_model_ref_v1beta1
8282
types.deployment_resource_pool = types.deployment_resource_pool_v1beta1
83+
types.deployment_resource_pool_service = (
84+
types.deployment_resource_pool_service_v1beta1
85+
)
8386
types.encryption_spec = types.encryption_spec_v1beta1
8487
types.endpoint = types.endpoint_v1beta1
8588
types.endpoint_service = types.endpoint_service_v1beta1
@@ -159,6 +162,9 @@
159162
if DEFAULT_VERSION == V1:
160163

161164
services.dataset_service_client = services.dataset_service_client_v1
165+
services.deployment_resource_pool_service_client = (
166+
services.deployment_resource_pool_service_client_v1
167+
)
162168
services.endpoint_service_client = services.endpoint_service_client_v1
163169
services.feature_online_store_admin_service_client = (
164170
services.feature_online_store_admin_service_client_v1
@@ -205,6 +211,8 @@
205211
types.dataset = types.dataset_v1
206212
types.dataset_service = types.dataset_service_v1
207213
types.deployed_model_ref = types.deployed_model_ref_v1
214+
types.deployment_resource_pool = types.deployment_resource_pool_v1
215+
types.deployment_resource_pool_service = types.deployment_resource_pool_service_v1
208216
types.encryption_spec = types.encryption_spec_v1
209217
types.endpoint = types.endpoint_v1
210218
types.endpoint_service = types.endpoint_service_v1

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

+4
Original file line numberDiff line numberDiff line change
@@ -116,6 +116,9 @@
116116
from google.cloud.aiplatform_v1.services.dataset_service import (
117117
client as dataset_service_client_v1,
118118
)
119+
from google.cloud.aiplatform_v1.services.deployment_resource_pool_service import (
120+
client as deployment_resource_pool_service_client_v1,
121+
)
119122
from google.cloud.aiplatform_v1.services.endpoint_service import (
120123
client as endpoint_service_client_v1,
121124
)
@@ -180,6 +183,7 @@
180183
__all__ = (
181184
# v1
182185
dataset_service_client_v1,
186+
deployment_resource_pool_service_client_v1,
183187
endpoint_service_client_v1,
184188
feature_online_store_service_client_v1,
185189
feature_online_store_admin_service_client_v1,

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

+4
Original file line numberDiff line numberDiff line change
@@ -124,6 +124,8 @@
124124
dataset_service as dataset_service_v1,
125125
deployed_index_ref as matching_engine_deployed_index_ref_v1,
126126
deployed_model_ref as deployed_model_ref_v1,
127+
deployment_resource_pool as deployment_resource_pool_v1,
128+
deployment_resource_pool_service as deployment_resource_pool_service_v1,
127129
encryption_spec as encryption_spec_v1,
128130
endpoint as endpoint_v1,
129131
endpoint_service as endpoint_service_v1,
@@ -206,6 +208,8 @@
206208
dataset_v1,
207209
dataset_service_v1,
208210
deployed_model_ref_v1,
211+
deployment_resource_pool_v1,
212+
deployment_resource_pool_service_v1,
209213
encryption_spec_v1,
210214
endpoint_v1,
211215
endpoint_service_v1,

0 commit comments

Comments
 (0)