diff --git a/coordinator/gscoordinator/cluster_builder.py b/coordinator/gscoordinator/cluster_builder.py index 4f3a8ab8455a..cdbb31ffb38f 100644 --- a/coordinator/gscoordinator/cluster_builder.py +++ b/coordinator/gscoordinator/cluster_builder.py @@ -481,7 +481,7 @@ def vineyard_service_name(self): def get_vineyard_service_endpoint(self, api_client): # return f"{self.vineyard_service_name}:{self._vineyard_service_port}" service_name = self.vineyard_service_name - service_type = "ClusterIP" + service_type = self._service_type if self.vineyard_deployment_exists(): service_name = self._vineyard_deployment + "-rpc" endpoints = get_service_endpoints( diff --git a/coordinator/gscoordinator/kubernetes_launcher.py b/coordinator/gscoordinator/kubernetes_launcher.py index 5abbcb8ffd7e..2c38931aa9ca 100644 --- a/coordinator/gscoordinator/kubernetes_launcher.py +++ b/coordinator/gscoordinator/kubernetes_launcher.py @@ -618,6 +618,7 @@ def _inject_vineyard_as_sidecar(self, workload): sidecar_size=self._vineyard_shared_mem, sidecar_cpu=self._vineyard_cpu, sidecar_memory=self._vineyard_mem, + sidecar_service_type=self._service_type, output="json", capture=True, )