fix: list workloads in k8s

thxCode · thxCode · commit 01b328b34819 · 2025-12-05T13:43:30.000+08:00
Signed-off-by: thxCode &lt;thxcode0824@gmail.com&gt;
diff --git a/gpustack_runtime/deployer/kuberentes.py b/gpustack_runtime/deployer/kuberentes.py
@@ -1697,14 +1697,14 @@ def _list(
 
         return [
             KubernetesWorkloadStatus(
-                name=k_pod.metadata.labels.get(_LABEL_WORKLOAD, k_pod.metadata.name),
+                name=k_pod.metadata.labels[_LABEL_WORKLOAD],
                 k_pod=k_pod,
             )
             for k_pod in k_pods.items or []
             if (
                 k_pod.metadata.labels
-                and (_LABEL_WORKLOAD in k_pod.metadata.labels)
-                and k_pod.metadata.labels[_LABEL_WORKLOAD] == k_pod.metadata.name
+                and _LABEL_WORKLOAD in k_pod.metadata.labels
+                and k_pod.spec.node_name == self._node_name
             )
         ]
 

Original file line number	Diff line number	Diff line change
`@@ -1697,14 +1697,14 @@ def _list(`
`1697`	`1697`
`1698`	`1698`	`return [`
`1699`	`1699`	`KubernetesWorkloadStatus(`
`1700`		`- name=k_pod.metadata.labels.get(_LABEL_WORKLOAD, k_pod.metadata.name),`
	`1700`	`+ name=k_pod.metadata.labels[_LABEL_WORKLOAD],`
`1701`	`1701`	`k_pod=k_pod,`
`1702`	`1702`	`)`
`1703`	`1703`	`for k_pod in k_pods.items or []`
`1704`	`1704`	`if (`
`1705`	`1705`	`k_pod.metadata.labels`
`1706`		`- and (_LABEL_WORKLOAD in k_pod.metadata.labels)`
`1707`		`- and k_pod.metadata.labels[_LABEL_WORKLOAD] == k_pod.metadata.name`
	`1706`	`+ and _LABEL_WORKLOAD in k_pod.metadata.labels`
	`1707`	`+ and k_pod.spec.node_name == self._node_name`
`1708`	`1708`	`)`
`1709`	`1709`	`]`
`1710`	`1710`