From 56925f1626af75417cb1fe427e43a4cba717184d Mon Sep 17 00:00:00 2001 From: MichaelClifford Date: Wed, 12 Apr 2023 14:15:13 -0400 Subject: [PATCH 1/2] add MY_POD_IP to head nodes --- src/codeflare_sdk/templates/new-template.yaml | 7 ++++++- tests/test-case-cmd.yaml | 7 ++++++- 2 files changed, 12 insertions(+), 2 deletions(-) diff --git a/src/codeflare_sdk/templates/new-template.yaml b/src/codeflare_sdk/templates/new-template.yaml index d8e923dc9..585f05baa 100644 --- a/src/codeflare_sdk/templates/new-template.yaml +++ b/src/codeflare_sdk/templates/new-template.yaml @@ -114,7 +114,12 @@ spec: - "aw-kuberay" containers: # The Ray head pod - - name: ray-head + - env: + - name: MY_POD_IP + valueFrom: + fieldRef: + fieldPath: status.podIP + name: ray-head image: rayproject/ray:latest imagePullPolicy: Always ports: diff --git a/tests/test-case-cmd.yaml b/tests/test-case-cmd.yaml index 6782c50f5..4e5d72534 100644 --- a/tests/test-case-cmd.yaml +++ b/tests/test-case-cmd.yaml @@ -57,7 +57,12 @@ spec: template: spec: containers: - - image: rayproject/ray:latest + - env: + - name: MY_POD_IP + valueFrom: + fieldRef: + fieldPath: status.podIP + image: rayproject/ray:latest imagePullPolicy: Always lifecycle: preStop: From 7bb3e64f06212092d53c49b7d7df378690c534b2 Mon Sep 17 00:00:00 2001 From: MichaelClifford Date: Wed, 12 Apr 2023 16:04:29 -0400 Subject: [PATCH 2/2] update test-case yamls --- tests/test-case.yaml | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/tests/test-case.yaml b/tests/test-case.yaml index bdaf1b722..2aebb0007 100644 --- a/tests/test-case.yaml +++ b/tests/test-case.yaml @@ -68,7 +68,12 @@ spec: values: - unit-test-cluster containers: - - image: ghcr.io/foundation-model-stack/base:ray2.1.0-py38-gpu-pytorch1.12.0cu116-20221213-193103 + - env: + - name: MY_POD_IP + valueFrom: + fieldRef: + fieldPath: status.podIP + image: ghcr.io/foundation-model-stack/base:ray2.1.0-py38-gpu-pytorch1.12.0cu116-20221213-193103 imagePullPolicy: Always lifecycle: preStop: