Title

Fix GPU count fetching and refactor node retrieval

Description

• Fixed bug in fetching GPU count from nodes

• Updated node retrieval method to use Get instead of List

• Corrected error handling and return statements

• Refactored variable names and imports for consistency

Changes walkthrough 📝

Need help?

Type /help how to ... in the comments thread for any questions about PR-Agent usage.

Check out the documentation for more information.

PR Reviewer Guide 🔍

Here are some key observations to aid the review process:

func GetPerNodeGPUCountFromNodes(nodeList *corev1.NodeList) int {
	for _, node := range nodeList.Items {
		gpuCount, exists := node.Status.Capacity[consts.NvidiaGPU]
		if exists && gpuCount.String() != "" {
			return int(gpuCount.Value())

if err := kubeClient.Get(ctx, types.NamespacedName{Name: nodeName}, node); err != nil { // Note: nodes don't have a namespace here.
	return 0, fmt.Errorf("failed to get node %s: %w", nodeName, err)
}
allNodes.Items = append(allNodes.Items, *node)

@Jont828 could you resolve the unit test failure?