From 9bf8f741ee32ebd861c59c64aec6c8eacc14ee45 Mon Sep 17 00:00:00 2001
From: yliao <yliao@google.com>
Date: Mon, 2 Feb 2026 19:08:17 +0000
Subject: [PATCH] fix test/integration/scheduler/batch

---
 .../plugins/noderesources/balanced_allocation.go         | 4 ----
 .../plugins/noderesources/balanced_allocation_test.go    | 6 ------
 pkg/scheduler/framework/plugins/noderesources/fit.go     | 3 ---
 .../framework/plugins/noderesources/fit_test.go          | 9 ---------
 4 files changed, 22 deletions(-)

diff --git a/pkg/scheduler/framework/plugins/noderesources/balanced_allocation.go b/pkg/scheduler/framework/plugins/noderesources/balanced_allocation.go
index 36041fb8d33..efd0bdc7e4e 100644
--- a/pkg/scheduler/framework/plugins/noderesources/balanced_allocation.go
+++ b/pkg/scheduler/framework/plugins/noderesources/balanced_allocation.go
@@ -124,10 +124,6 @@ func (pl *BalancedAllocation) SignPod(ctx context.Context, pod *v1.Pod) ([]fwk.S
 		EnableDRAExtendedResource: pl.enableDRAExtendedResource,
 	}
 
-	if pl.enableDRAExtendedResource {
-		return nil, fwk.NewStatus(fwk.Unschedulable, "signature disabled when dra extended resources enabled")
-	}
-
 	return []fwk.SignFragment{
 		{Key: fwk.ResourcesSignerName, Value: computePodResourceRequest(pod, opts)},
 	}, nil
diff --git a/pkg/scheduler/framework/plugins/noderesources/balanced_allocation_test.go b/pkg/scheduler/framework/plugins/noderesources/balanced_allocation_test.go
index 23346916d84..b97938650dd 100644
--- a/pkg/scheduler/framework/plugins/noderesources/balanced_allocation_test.go
+++ b/pkg/scheduler/framework/plugins/noderesources/balanced_allocation_test.go
@@ -395,12 +395,6 @@ func TestBalancedAllocationSignPod(t *testing.T) {
 			},
 			expectedStatusCode: fwk.Success,
 		},
-		"DRA extended resource enabled - returns unschedulable": {
-			pod:                       st.MakePod().Req(cpuAndMemory("1000m", "2000")).Obj(),
-			enableDRAExtendedResource: true,
-			expectedFragments:         nil,
-			expectedStatusCode:        fwk.Unschedulable,
-		},
 	}
 
 	for name, test := range tests {
diff --git a/pkg/scheduler/framework/plugins/noderesources/fit.go b/pkg/scheduler/framework/plugins/noderesources/fit.go
index f20749e38d9..aa04bdcf24b 100644
--- a/pkg/scheduler/framework/plugins/noderesources/fit.go
+++ b/pkg/scheduler/framework/plugins/noderesources/fit.go
@@ -176,9 +176,6 @@ func (pl *Fit) SignPod(ctx context.Context, pod *v1.Pod) ([]fwk.SignFragment, *f
 		EnablePodLevelResources:   pl.enablePodLevelResources,
 		EnableDRAExtendedResource: pl.enableDRAExtendedResource,
 	}
-	if pl.enableDRAExtendedResource {
-		return nil, fwk.NewStatus(fwk.Unschedulable, "signature disabled when dra extended resources enabled")
-	}
 	return []fwk.SignFragment{
 		{Key: fwk.ResourcesSignerName, Value: computePodResourceRequest(pod, opts)},
 	}, nil
diff --git a/pkg/scheduler/framework/plugins/noderesources/fit_test.go b/pkg/scheduler/framework/plugins/noderesources/fit_test.go
index 9e07442b2ef..84aabdcb714 100644
--- a/pkg/scheduler/framework/plugins/noderesources/fit_test.go
+++ b/pkg/scheduler/framework/plugins/noderesources/fit_test.go
@@ -2222,15 +2222,6 @@ func TestFitSignPod(t *testing.T) {
 			},
 			expectedStatusCode: fwk.Success,
 		},
-		"DRA extended resource enabled - returns unschedulable": {
-			pod: st.MakePod().Req(map[v1.ResourceName]string{
-				v1.ResourceCPU:    "1000m",
-				v1.ResourceMemory: "2000",
-			}).Obj(),
-			enableDRAExtendedResource: true,
-			expectedFragments:         nil,
-			expectedStatusCode:        fwk.Unschedulable,
-		},
 	}
 
 	for name, test := range tests {