Changed serviceName to match new headless-svc

GoogleCloudPlatform · Nov 22, 2023 · 90adb13 · 90adb13
1 parent 385b0f3
commit 90adb13
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/ray-on-gke/user/kuberay-tpu-webhook/bin/kuberay-tpu-webhook b/ray-on-gke/user/kuberay-tpu-webhook/bin/kuberay-tpu-webhook
diff --git a/ray-on-gke/user/kuberay-tpu-webhook/main.go b/ray-on-gke/user/kuberay-tpu-webhook/main.go
@@ -92,10 +92,9 @@ func mutateRayCluster(
 			topology := raycluster.Spec.WorkerGroupSpecs[i].Template.Spec.NodeSelector["cloud.google.com/gke-tpu-topology"]
 			if(isTPUMultiHost(topology)) {
 				numWorkers := int(*raycluster.Spec.WorkerGroupSpecs[i].Replicas)
-
 				// generate DNS hostnames
 				workerGroupName := raycluster.Spec.WorkerGroupSpecs[i].GroupName
-				serviceName := "headless-svc" // TODO: get headless-svc associated with workergroup
+				serviceName := "tpu-worker-group-svc" // TODO: get headless-svc associated with workergroup
 				hostNames := make([]string, numWorkers)
 				for j := 0; j < numWorkers; j++ {
 					hostNames[j] = fmt.Sprintf("%s-%d.%s", workerGroupName, j, serviceName)