PR fixes

run-ai · Jul 16, 2024 · 94cab57 · 94cab57
1 parent bad9106
commit 94cab57
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 5 deletions.
diff --git a/deploy/fake-gpu-operator/values.yaml b/deploy/fake-gpu-operator/values.yaml
@@ -72,4 +72,4 @@ topology:
       gpuCount: 2
       gpuMemory: 11441
   nodePoolLabelKey: run.ai/simulated-gpu-node-pool
-  mig-strategy: mixed
+  migStrategy: mixed
diff --git a/internal/status-updater/handlers/node/labels.go b/internal/status-updater/handlers/node/labels.go
@@ -10,11 +10,16 @@ import (
 	"k8s.io/apimachinery/pkg/types"
 )
 
+const (
+	dcgmExporterLabelKey = "nvidia.com/gpu.deploy.dcgm-exporter"
+	devicePluginLabelKey = "nvidia.com/gpu.deploy.device-plugin"
+)
+
 // 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{}{
-		"nvidia.com/gpu.deploy.dcgm-exporter": "true",
-		"nvidia.com/gpu.deploy.device-plugin": "true",
+		dcgmExporterLabelKey: "true",
+		devicePluginLabelKey: "true",
 	})
 	if err != nil {
 		return fmt.Errorf("failed to label node %s: %w", node.Name, err)
@@ -26,8 +31,8 @@ func (p *NodeHandler) labelNode(node *v1.Node) error {
 // unlabelNode removes the labels from the node that were added by the fake-gpu-operator.
 func (p *NodeHandler) unlabelNode(node *v1.Node) error {
 	err := p.patchNodeLabels(node, map[string]interface{}{
-		"nvidia.com/gpu.deploy.dcgm-exporter": nil,
-		"nvidia.com/gpu.deploy.device-plugin": nil,
+		dcgmExporterLabelKey: nil,
+		devicePluginLabelKey: nil,
 	})
 	if err != nil {
 		return fmt.Errorf("failed to unlabel node %s: %w", node.Name, err)