Update pkg/scheduler/webhook_test.go

Signed-off-by: Harry Li <[email protected]>

Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>

Author: mayooot

pkg/scheduler/webhook_test.go

@@ -245,21 +245,21 @@ func TestPodHasDifferentScheduler(t *testing.T) {
 func TestPodHasInitContainer(t *testing.T) {
 	config.SchedulerName = "hami-scheduler"
 	config.ForceOverwriteDefaultScheduler = true
-	config := &device.Config{
-		NvidiaConfig: nvidia.NvidiaConfig{
-			ResourceCountName:            "hami.io/gpu",
-			ResourceMemoryName:           "hami.io/gpumem",
-			ResourceMemoryPercentageName: "hami.io/gpumem-percentage",
-			ResourceCoreName:             "hami.io/gpucores",
-			DefaultMemory:                0,
-			DefaultCores:                 0,
-			DefaultGPUNum:                1,
-		},
-	}
-
-	if err := device.InitDevicesWithConfig(config); err != nil {
-		klog.Fatalf("Failed to initialize devices with config: %v", err)
-	}
+ 	deviceConfig := &device.Config{
+ 		NvidiaConfig: nvidia.NvidiaConfig{
+ 			ResourceCountName:            "hami.io/gpu",
+ 			ResourceMemoryName:           "hami.io/gpumem",
+ 			ResourceMemoryPercentageName: "hami.io/gpumem-percentage",
+ 			ResourceCoreName:             "hami.io/gpucores",
+ 			DefaultMemory:                0,
+ 			DefaultCores:                 0,
+ 			DefaultGPUNum:                1,
+ 		},
+ 	}
+ 
+ 	if err := device.InitDevicesWithConfig(deviceConfig); err != nil {
+ 		klog.Fatalf("Failed to initialize devices with config: %v", err)
+ 	}
 
 	pod := &corev1.Pod{
 		ObjectMeta: metav1.ObjectMeta{