RUN-20790 Don't run device plugin DaemonSet on fake nodes (#102)

internal/status-updater/handlers/node/common.go

@@ -0,0 +1,10 @@
+package node
+
+import (
+	"github.com/run-ai/fake-gpu-operator/internal/common/constants"
+	v1 "k8s.io/api/core/v1"
+)
+
+func isFakeNode(node *v1.Node) bool {
+	return node != nil && node.Annotations[constants.AnnotationKwokNode] == "fake"
+}

internal/status-updater/handlers/node/fake_node_deployments.go

@@ -119,7 +119,3 @@ func generateFakeNodeDeploymentFromTemplate(template *appsv1.Deployment, node *v
 
 	return deployment
 }
-
-func isFakeNode(node *v1.Node) bool {
-	return node != nil && node.Annotations[constants.AnnotationKwokNode] == "fake"
-}

internal/status-updater/handlers/node/labels.go

@@ -18,10 +18,14 @@ const (
 
 // labelNode labels the node with required labels for the fake-gpu-operator to function.
 func (p *NodeHandler) labelNode(node *v1.Node) error {
-	err := p.patchNodeLabels(node, map[string]interface{}{
+	labels := map[string]interface{}{
 		dcgmExporterLabelKey: "true",
-		devicePluginLabelKey: "true",
-	})
+	}
+	if !isFakeNode(node) {
+		labels[devicePluginLabelKey] = "true"
+	}
+
+	err := p.patchNodeLabels(node, labels)
 	if err != nil {
 		return fmt.Errorf("failed to label node %s: %w", node.Name, err)
 	}