Skip to content

Commit

Permalink
feat: Migrate DeploymentResourcePool and associated functionality to …
Browse files Browse the repository at this point in the history
…V1 namespace for GA launch of model co-hosting.

PiperOrigin-RevId: 649164615
  • Loading branch information
vertex-sdk-bot authored and copybara-github committed Jul 3, 2024
1 parent b63f960 commit 1474d98
Show file tree
Hide file tree
Showing 9 changed files with 785 additions and 109 deletions.
8 changes: 8 additions & 0 deletions google/cloud/aiplatform/compat/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -80,6 +80,9 @@
types.dataset_service = types.dataset_service_v1beta1
types.deployed_model_ref = types.deployed_model_ref_v1beta1
types.deployment_resource_pool = types.deployment_resource_pool_v1beta1
types.deployment_resource_pool_service = (
types.deployment_resource_pool_service_v1beta1
)
types.encryption_spec = types.encryption_spec_v1beta1
types.endpoint = types.endpoint_v1beta1
types.endpoint_service = types.endpoint_service_v1beta1
Expand Down Expand Up @@ -159,6 +162,9 @@
if DEFAULT_VERSION == V1:

services.dataset_service_client = services.dataset_service_client_v1
services.deployment_resource_pool_service_client = (
services.deployment_resource_pool_service_client_v1
)
services.endpoint_service_client = services.endpoint_service_client_v1
services.feature_online_store_admin_service_client = (
services.feature_online_store_admin_service_client_v1
Expand Down Expand Up @@ -205,6 +211,8 @@
types.dataset = types.dataset_v1
types.dataset_service = types.dataset_service_v1
types.deployed_model_ref = types.deployed_model_ref_v1
types.deployment_resource_pool = types.deployment_resource_pool_v1
types.deployment_resource_pool_service = types.deployment_resource_pool_service_v1
types.encryption_spec = types.encryption_spec_v1
types.endpoint = types.endpoint_v1
types.endpoint_service = types.endpoint_service_v1
Expand Down
4 changes: 4 additions & 0 deletions google/cloud/aiplatform/compat/services/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -116,6 +116,9 @@
from google.cloud.aiplatform_v1.services.dataset_service import (
client as dataset_service_client_v1,
)
from google.cloud.aiplatform_v1.services.deployment_resource_pool_service import (
client as deployment_resource_pool_service_client_v1,
)
from google.cloud.aiplatform_v1.services.endpoint_service import (
client as endpoint_service_client_v1,
)
Expand Down Expand Up @@ -180,6 +183,7 @@
__all__ = (
# v1
dataset_service_client_v1,
deployment_resource_pool_service_client_v1,
endpoint_service_client_v1,
feature_online_store_service_client_v1,
feature_online_store_admin_service_client_v1,
Expand Down
4 changes: 4 additions & 0 deletions google/cloud/aiplatform/compat/types/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -124,6 +124,8 @@
dataset_service as dataset_service_v1,
deployed_index_ref as matching_engine_deployed_index_ref_v1,
deployed_model_ref as deployed_model_ref_v1,
deployment_resource_pool as deployment_resource_pool_v1,
deployment_resource_pool_service as deployment_resource_pool_service_v1,
encryption_spec as encryption_spec_v1,
endpoint as endpoint_v1,
endpoint_service as endpoint_service_v1,
Expand Down Expand Up @@ -206,6 +208,8 @@
dataset_v1,
dataset_service_v1,
deployed_model_ref_v1,
deployment_resource_pool_v1,
deployment_resource_pool_service_v1,
encryption_spec_v1,
endpoint_v1,
endpoint_service_v1,
Expand Down
Loading

0 comments on commit 1474d98

Please sign in to comment.