adapt to new PreScore() interface

kubernetes-sigs · Aug 18, 2024 · 67300af · 67300af
1 parent 6022686
commit 67300af
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/pkg/trimaran/lowriskovercommitment/lowriskovercommitment.go b/pkg/trimaran/lowriskovercommitment/lowriskovercommitment.go
@@ -89,7 +89,7 @@ func New(_ context.Context, obj runtime.Object, handle framework.Handle) (framew
 }
 
 // PreScore : calculate pod requests and limits and store as plugin state data to be used during scoring
-func (pl *LowRiskOverCommitment) PreScore(ctx context.Context, cycleState *framework.CycleState, pod *v1.Pod, nodes []*v1.Node) *framework.Status {
+func (pl *LowRiskOverCommitment) PreScore(ctx context.Context, cycleState *framework.CycleState, pod *v1.Pod, nodes []*framework.NodeInfo) *framework.Status {
 	klog.V(6).InfoS("PreScore: Calculating pod resource requests and limits", "pod", klog.KObj(pod))
 	podResourcesStateData := CreatePodResourcesStateData(pod)
 	cycleState.Write(PodResourcesKey, podResourcesStateData)

diff --git a/pkg/trimaran/lowriskovercommitment/lowriskovercommitment_test.go b/pkg/trimaran/lowriskovercommitment/lowriskovercommitment_test.go
@@ -217,7 +217,7 @@ func TestLowRiskOverCommitment_Score(t *testing.T) {
 			p, _ := New(ctx, &lowRiskOverCommitmentArgs, fh)
 
 			preScorePlugin := p.(framework.PreScorePlugin)
-			status := preScorePlugin.PreScore(context.Background(), state, tt.pod, tt.nodes)
+			status := preScorePlugin.PreScore(context.Background(), state, tt.pod, tf.BuildNodeInfos(tt.nodes))
 			assert.True(t, status.IsSuccess())
 
 			scorePlugin := p.(framework.ScorePlugin)