Update gpu_usage_calculator.go

eliranw · web-flow · commit 8d0a362e423e · 2025-12-21T16:53:34.000+02:00
diff --git a/internal/status-updater/handlers/pod/gpu_usage_calculator.go b/internal/status-updater/handlers/pod/gpu_usage_calculator.go
@@ -104,9 +104,7 @@ func calculateUtilizationFromAnnotation(annotationValue string) (*topology.Range
 }
 
 func getPodType(dynamicClient dynamic.Interface, pod *v1.Pod) (string, error) {
-	// First, try to get workload type from pod labels
 	if workloadKind, ok := pod.Labels["workloadKind"]; ok {
-		// Map Run:AI workload kinds to utilization types
 		switch workloadKind {
 		case "TrainingWorkload":
 			return "train", nil

Original file line number	Diff line number	Diff line change
`@@ -104,9 +104,7 @@ func calculateUtilizationFromAnnotation(annotationValue string) (*topology.Range`
`104`	`104`	`}`
`105`	`105`
`106`	`106`	`func getPodType(dynamicClient dynamic.Interface, pod *v1.Pod) (string, error) {`
`107`		`- // First, try to get workload type from pod labels`
`108`	`107`	`if workloadKind, ok := pod.Labels["workloadKind"]; ok {`
`109`		`- // Map Run:AI workload kinds to utilization types`
`110`	`108`	`switch workloadKind {`
`111`	`109`	`case "TrainingWorkload":`
`112`	`110`	`return "train", nil`