Retain coverage

Signed-off-by: Jason Parraga <[email protected]>
flyteorg · Feb 8, 2025 · 408624e · 408624e
1 parent 6b06ff6
commit 408624e
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/flyteplugins/go/tasks/plugins/k8s/ray/ray_test.go b/flyteplugins/go/tasks/plugins/k8s/ray/ray_test.go
@@ -590,10 +590,12 @@ func TestBuildResourceRayCustomK8SPod(t *testing.T) {
 			},
 			headPodAssertions: rayPodAssertions{
 				affinity:         &corev1.Affinity{},
+				resources:        resourceRequirements,
 				runtimeClassName: &nvidiaRuntimeClassName,
 			},
 			workerPodAssertions: rayPodAssertions{
 				affinity:         &corev1.Affinity{},
+				resources:        resourceRequirements,
 				runtimeClassName: &nvidiaRuntimeClassName,
 			},
 		},
@@ -607,10 +609,12 @@ func TestBuildResourceRayCustomK8SPod(t *testing.T) {
 			},
 			headPodAssertions: rayPodAssertions{
 				affinity:    &corev1.Affinity{},
+				resources:   resourceRequirements,
 				tolerations: headTolerations,
 			},
 			workerPodAssertions: rayPodAssertions{
 				affinity:    &corev1.Affinity{},
+				resources:   resourceRequirements,
 				tolerations: workerTolerations,
 			},
 		},
@@ -627,10 +631,12 @@ func TestBuildResourceRayCustomK8SPod(t *testing.T) {
 				}),
 			},
 			headPodAssertions: rayPodAssertions{
-				affinity: headAffinity,
+				affinity:  headAffinity,
+				resources: resourceRequirements,
 			},
 			workerPodAssertions: rayPodAssertions{
-				affinity: workerAffinity,
+				affinity:  workerAffinity,
+				resources: resourceRequirements,
 			},
 		},
 	}