Skip to content

Commit da76253

Browse files
yinghsienwucopybara-github
authored andcommitted
feat: Support PSC-I ingress in Ray Client
PiperOrigin-RevId: 688719900
1 parent f713417 commit da76253

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

google/cloud/aiplatform/vertex_ray/client_builder.py

+4-1
Original file line numberDiff line numberDiff line change
@@ -177,7 +177,10 @@ def connect(self) -> _VertexRayClientContext:
177177
public_address = self.response.resource_runtime.access_uris.get(
178178
"RAY_CLIENT_ENDPOINT"
179179
)
180-
if public_address:
180+
private_address = self.response.resource_runtime.access_uris.get(
181+
"RAY_HEAD_NODE_INTERNAL_IP"
182+
)
183+
if public_address and not private_address:
181184
self._credentials = grpc.ssl_channel_credentials()
182185
bearer_token = _validation_utils.get_bearer_token()
183186
self._metadata = [

0 commit comments

Comments
 (0)