extend PodSets() to compute TopologyRequest

dgrove-oss · dgrove-oss · commit 7983bbace636 · 2025-01-14T17:26:33.000-05:00
diff --git a/internal/controller/workload/workload_controller.go b/internal/controller/workload/workload_controller.go
@@ -77,6 +77,10 @@ func (aw *AppWrapper) PodSets() []kueue.PodSet {
 		// Kueue will raise an error on zero length PodSet; the Kueue GenericJob API prevents propagating the actual error.
 		return []kueue.PodSet{}
 	}
+	for psIndex := range podSets {
+		podSets[psIndex].TopologyRequest = jobframework.PodSetTopologyRequest(&podSets[psIndex].Template.ObjectMeta, nil, nil, nil)
+	}
+
 	return podSets
 }
 

Original file line number	Diff line number	Diff line change
`@@ -77,6 +77,10 @@ func (aw *AppWrapper) PodSets() []kueue.PodSet {`
`77`	`77`	`// Kueue will raise an error on zero length PodSet; the Kueue GenericJob API prevents propagating the actual error.`
`78`	`78`	`return []kueue.PodSet{}`
`79`	`79`	`}`
	`80`	`+ for psIndex := range podSets {`
	`81`	`+ podSets[psIndex].TopologyRequest = jobframework.PodSetTopologyRequest(&podSets[psIndex].Template.ObjectMeta, nil, nil, nil)`
	`82`	`+ }`
	`83`	`+`
`80`	`84`	`return podSets`
`81`	`85`	`}`
`82`	`86`