From 4500c19e9b36cd51d9539d7fd594b841b2c8f4ed Mon Sep 17 00:00:00 2001 From: Varun Gupta Date: Sun, 2 Mar 2025 18:54:10 -0800 Subject: [PATCH] ignore worker pods in UpdatePod as well Signed-off-by: Varun Gupta --- pkg/cache/cache.go | 15 +++++++++++++++ 1 file changed, 15 insertions(+) diff --git a/pkg/cache/cache.go b/pkg/cache/cache.go index ddc878f6..dab8a60e 100644 --- a/pkg/cache/cache.go +++ b/pkg/cache/cache.go @@ -300,6 +300,7 @@ func (c *Cache) addPod(obj interface{}) { // ignore worker pods nodeType, ok := pod.Labels[nodeType] if ok && nodeType == "worker" { + klog.InfoS("ignored ray worker pod", "name", pod.Name) return } @@ -329,6 +330,20 @@ func (c *Cache) updatePod(oldObj interface{}, newObj interface{}) { c.deletePodAndModelMapping(oldPod.Name, oldModelName) } + // ignore worker pods + nodeType, ok := oldPod.Labels[nodeType] + if ok && nodeType == "worker" { + klog.InfoS("ignored ray worker pod", "name", oldPod.Name) + return + } + + // ignore worker pods + nodeType, ok = newPod.Labels[nodeType] + if ok && nodeType == "worker" { + klog.InfoS("ignored ray worker pod", "name", newPod.Name) + return + } + // Add new mappings if present if newOk { c.Pods[newPod.Name] = newPod