From 2b6eb4c07f30740b0e4c92b05c5b8369bb7a850d Mon Sep 17 00:00:00 2001 From: Varun Gupta Date: Sun, 2 Mar 2025 22:19:28 -0800 Subject: [PATCH] use node worker as const Signed-off-by: Varun Gupta --- pkg/cache/cache.go | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/pkg/cache/cache.go b/pkg/cache/cache.go index dab8a60e..d59358ff 100644 --- a/pkg/cache/cache.go +++ b/pkg/cache/cache.go @@ -76,6 +76,7 @@ type Block struct { const ( modelIdentifier = "model.aibrix.ai/name" nodeType = "ray.io/node-type" + nodeWorker = "worker" podPort = 8000 defaultPodMetricRefreshIntervalInMS = 50 expireWriteRequestTraceIntervalInMins = 10 @@ -299,7 +300,7 @@ func (c *Cache) addPod(obj interface{}) { } // ignore worker pods nodeType, ok := pod.Labels[nodeType] - if ok && nodeType == "worker" { + if ok && nodeType == nodeWorker { klog.InfoS("ignored ray worker pod", "name", pod.Name) return } @@ -332,14 +333,14 @@ func (c *Cache) updatePod(oldObj interface{}, newObj interface{}) { // ignore worker pods nodeType, ok := oldPod.Labels[nodeType] - if ok && nodeType == "worker" { + if ok && nodeType == nodeWorker { klog.InfoS("ignored ray worker pod", "name", oldPod.Name) return } // ignore worker pods nodeType, ok = newPod.Labels[nodeType] - if ok && nodeType == "worker" { + if ok && nodeType == nodeWorker { klog.InfoS("ignored ray worker pod", "name", newPod.Name) return }