From dd4b8529a72f6d08e56e8437cdd3cd0874df8a47 Mon Sep 17 00:00:00 2001 From: Matthew Tang Date: Wed, 15 Nov 2023 17:10:15 -0800 Subject: [PATCH] feat: Populate Ray Cluster dashboard_address from proto field PiperOrigin-RevId: 582853171 --- .../cloud/aiplatform/preview/vertex_ray/util/_gapic_utils.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/google/cloud/aiplatform/preview/vertex_ray/util/_gapic_utils.py b/google/cloud/aiplatform/preview/vertex_ray/util/_gapic_utils.py index fc61fb794a..93e6a1f12d 100644 --- a/google/cloud/aiplatform/preview/vertex_ray/util/_gapic_utils.py +++ b/google/cloud/aiplatform/preview/vertex_ray/util/_gapic_utils.py @@ -138,11 +138,15 @@ def persistent_resource_to_cluster( Returns: Cluster. """ + dashboard_address = persistent_resource.resource_runtime.access_uris.get( + "RAY_DASHBOARD_URI" + ) cluster = Cluster( cluster_resource_name=persistent_resource.name, network=persistent_resource.network, state=persistent_resource.state.name, labels=persistent_resource.labels, + dashboard_address=dashboard_address, ) if not persistent_resource.resource_runtime_spec.ray_spec: # skip PersistentResource without RaySpec