address rabbit comments

XRFXLP · XRFXLP · commit abbbaffad588 · 2026-06-23T09:11:48.000+05:30
Signed-off-by: Ajay Mishra &lt;ajmishra@nvidia.com&gt;
diff --git a/labeler/main.go b/labeler/main.go
@@ -152,5 +152,10 @@ func loadExpectedDeviceCountsConfig(configFile string) (devicecounts.Config, err
 		return devicecounts.Config{}, nil
 	}
 
-	return devicecounts.LoadConfig(configFile)
+	config, err := devicecounts.LoadConfig(configFile)
+	if err != nil {
+		return devicecounts.Config{}, fmt.Errorf("load expected device counts config %q: %w", configFile, err)
+	}
+
+	return config, nil
 }
diff --git a/labeler/pkg/devicecounts/device_counts.go b/labeler/pkg/devicecounts/device_counts.go
@@ -144,6 +144,10 @@ func (m *Manager) ReconcileNodeLabelsInPlace(
 		return false
 	}
 
+	if node.Labels == nil {
+		node.Labels = make(map[string]string)
+	}
+
 	needsUpdate := false
 	resourceSlices := resourceSlicesForNode(node)
 

Original file line number	Diff line number	Diff line change
`@@ -152,5 +152,10 @@ func loadExpectedDeviceCountsConfig(configFile string) (devicecounts.Config, err`
`152`	`152`	`return devicecounts.Config{}, nil`
`153`	`153`	`}`
`154`	`154`
`155`		`- return devicecounts.LoadConfig(configFile)`
	`155`	`+ config, err := devicecounts.LoadConfig(configFile)`
	`156`	`+ if err != nil {`
	`157`	`+ return devicecounts.Config{}, fmt.Errorf("load expected device counts config %q: %w", configFile, err)`
	`158`	`+ }`
	`159`	`+`
	`160`	`+ return config, nil`
`156`	`161`	`}`
Original file line number	Diff line number	Diff line change
`@@ -144,6 +144,10 @@ func (m *Manager) ReconcileNodeLabelsInPlace(`
`144`	`144`	`return false`
`145`	`145`	`}`
`146`	`146`
	`147`	`+ if node.Labels == nil {`
	`148`	`+ node.Labels = make(map[string]string)`
	`149`	`+ }`
	`150`	`+`
`147`	`151`	`needsUpdate := false`
`148`	`152`	`resourceSlices := resourceSlicesForNode(node)`
`149`	`153`