fix unit test

Signed-off-by: You-Cheng Lin (Owen) <[email protected]>

Author: owenowenisme

ray-operator/controllers/ray/batchscheduler/yunikorn/yunikorn_scheduler_test.go

@@ -317,6 +317,7 @@ func TestPropagateTaskGroupsAnnotationToRayClusterAndSubmitterPodTemplate(t *tes
 		corev1.ResourceCPU:    resource.MustParse("2"),
 		corev1.ResourceMemory: resource.MustParse("10Gi"),
 	})
+	rayCluster.Spec.WorkerGroupSpecs[0].NumOfHosts = 1
 
 	rayJobWithGangScheduling := createRayJobWithLabels(
 		"ray-job-with-gang-scheduling",
@@ -426,6 +427,7 @@ func TestAddMetadataToChildResourceFromRayCluster(t *testing.T) {
 			utils.RayGangSchedulingEnabled: "true",
 		},
 	)
+
 	addHeadPodSpec(rayCluster, corev1.ResourceList{
 		corev1.ResourceCPU:    resource.MustParse("1"),
 		corev1.ResourceMemory: resource.MustParse("1Gi"),
@@ -434,7 +436,7 @@ func TestAddMetadataToChildResourceFromRayCluster(t *testing.T) {
 		corev1.ResourceCPU:    resource.MustParse("1"),
 		corev1.ResourceMemory: resource.MustParse("1Gi"),
 	})
-
+	rayCluster.Spec.WorkerGroupSpecs[0].NumOfHosts = 1
 	rayPod = createPod("ray-pod", "default")
 	yk.AddMetadataToChildResource(ctx, rayCluster, rayPod, "ray-cluster-with-gang-scheduling")
 
@@ -490,7 +492,7 @@ func TestAddMetadataToChildResourceFromRayJob(t *testing.T) {
 		corev1.ResourceCPU:    resource.MustParse("1"),
 		corev1.ResourceMemory: resource.MustParse("1Gi"),
 	})
-
+	rayCluster.Spec.WorkerGroupSpecs[0].NumOfHosts = 1
 	rayJob = createRayJobWithLabels(
 		"ray-job-with-gang-scheduling",
 		"test-namespace",