Add withSubmitterPodTemplate to spec in rayjob_lightweight_test

ray-project · May 2, 2024 · 49a8c90 · 49a8c90
1 parent e4d3cec
commit 49a8c90
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/ray-operator/test/e2e/rayjob_lightweight_test.go b/ray-operator/test/e2e/rayjob_lightweight_test.go
@@ -85,12 +85,13 @@ env_vars:
 	})
 
 	test.T().Run("Failing RayJob without cluster shutdown after finished", func(t *testing.T) {
-		rayJobAC := rayv1ac.RayJob("fail", namespace.Name).
+		rayJobAC := rayv1ac.RayJob("fail-k8s-job", namespace.Name).
 			WithSpec(rayv1ac.RayJobSpec().
 				WithSubmissionMode(rayv1.HTTPMode).
+				WithRayClusterSpec(newRayClusterSpec(mountConfigMap[rayv1ac.RayClusterSpecApplyConfiguration](jobs, "/home/ray/jobs"))).
 				WithEntrypoint("python /home/ray/jobs/fail.py").
 				WithShutdownAfterJobFinishes(false).
-				WithRayClusterSpec(newRayClusterSpec(mountConfigMap[rayv1ac.RayClusterSpecApplyConfiguration](jobs, "/home/ray/jobs"))))
+				WithSubmitterPodTemplate(jobSubmitterPodTemplateApplyConfiguration()))
 
 		rayJob, err := test.Client().Ray().RayV1().RayJobs(namespace.Name).Apply(test.Ctx(), rayJobAC, TestApplyOptions)
 		test.Expect(err).NotTo(HaveOccurred())