wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-06-28 20:56:34 +08:00
Code Issues Packages Projects Releases Wiki Activity

Replace read_namespaced_pod_status with read_namespaced_pod (#11278)

Browse Source
This commit is contained in:
Gekho457
2020-10-09 20:21:39 -04:00
committed by GitHub
parent 5ef1784024
commit 92b4059cad
2 changed files with 10 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/autoscaler/_private/kubernetes/node_provider.py
+5 -5
View File
@@ -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)
python/ray/autoscaler/_private/staroid/node_provider.py
+5 -5
View File
@@ -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)