diff --git a/python/ray/autoscaler/_private/kubernetes/node_provider.py b/python/ray/autoscaler/_private/kubernetes/node_provider.py index bed285a4b..aac45c904 100644 --- a/python/ray/autoscaler/_private/kubernetes/node_provider.py +++ b/python/ray/autoscaler/_private/kubernetes/node_provider.py @@ -48,22 +48,22 @@ class KubernetesNodeProvider(NodeProvider): return [pod.metadata.name for pod in pod_list.items] def is_running(self, node_id): - pod = core_api().read_namespaced_pod_status(node_id, self.namespace) + pod = core_api().read_namespaced_pod(node_id, self.namespace) return pod.status.phase == "Running" def is_terminated(self, node_id): - pod = core_api().read_namespaced_pod_status(node_id, self.namespace) + pod = core_api().read_namespaced_pod(node_id, self.namespace) return pod.status.phase not in ["Running", "Pending"] def node_tags(self, node_id): - pod = core_api().read_namespaced_pod_status(node_id, self.namespace) + pod = core_api().read_namespaced_pod(node_id, self.namespace) return pod.metadata.labels def external_ip(self, node_id): raise NotImplementedError("Must use internal IPs with Kubernetes.") def internal_ip(self, node_id): - pod = core_api().read_namespaced_pod_status(node_id, self.namespace) + pod = core_api().read_namespaced_pod(node_id, self.namespace) return pod.status.pod_ip def get_node_id(self, ip_address, use_internal_ip=True) -> str: @@ -72,7 +72,7 @@ class KubernetesNodeProvider(NodeProvider): return super().get_node_id(ip_address, use_internal_ip=use_internal_ip) def set_node_tags(self, node_id, tags): - pod = core_api().read_namespaced_pod_status(node_id, self.namespace) + pod = core_api().read_namespaced_pod(node_id, self.namespace) pod.metadata.labels.update(tags) core_api().patch_namespaced_pod(node_id, self.namespace, pod) diff --git a/python/ray/autoscaler/_private/staroid/node_provider.py b/python/ray/autoscaler/_private/staroid/node_provider.py index cde57fc3a..3afc93e0f 100644 --- a/python/ray/autoscaler/_private/staroid/node_provider.py +++ b/python/ray/autoscaler/_private/staroid/node_provider.py @@ -190,21 +190,21 @@ class StaroidNodeProvider(NodeProvider): kube_client = self.__cached[self.cluster_name]["kube_client"] core_api = client.CoreV1Api(kube_client) - pod = core_api.read_namespaced_pod_status(node_id, self.namespace) + pod = core_api.read_namespaced_pod(node_id, self.namespace) return pod.status.phase == "Running" def is_terminated(self, node_id): kube_client = self.__cached[self.cluster_name]["kube_client"] core_api = client.CoreV1Api(kube_client) - pod = core_api.read_namespaced_pod_status(node_id, self.namespace) + pod = core_api.read_namespaced_pod(node_id, self.namespace) return pod.status.phase not in ["Running", "Pending"] def node_tags(self, node_id): kube_client = self.__cached[self.cluster_name]["kube_client"] core_api = client.CoreV1Api(kube_client) - pod = core_api.read_namespaced_pod_status(node_id, self.namespace) + pod = core_api.read_namespaced_pod(node_id, self.namespace) return pod.metadata.labels def external_ip(self, node_id): @@ -214,7 +214,7 @@ class StaroidNodeProvider(NodeProvider): kube_client = self.__cached[self.cluster_name]["kube_client"] core_api = client.CoreV1Api(kube_client) - pod = core_api.read_namespaced_pod_status(node_id, self.namespace) + pod = core_api.read_namespaced_pod(node_id, self.namespace) return pod.status.pod_ip def get_node_id(self, ip_address, use_internal_ip=True) -> str: @@ -226,7 +226,7 @@ class StaroidNodeProvider(NodeProvider): kube_client = self.__cached[self.cluster_name]["kube_client"] core_api = client.CoreV1Api(kube_client) - pod = core_api.read_namespaced_pod_status(node_id, self.namespace) + pod = core_api.read_namespaced_pod(node_id, self.namespace) pod.metadata.labels.update(tags) core_api.patch_namespaced_pod(node_id, self.namespace, pod)