fix: gpu label test v0.37.x (#7156)

aws · Oct 3, 2024 · 7397a62 · 7397a62
1 parent 780ca37
commit 7397a62
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/test/suites/integration/scheduling_test.go b/test/suites/integration/scheduling_test.go
@@ -214,6 +214,7 @@ var _ = Describe("Scheduling", Ordered, ContinueOnFailure, func() {
 			env.ExpectCreatedNodeCount("==", 1)
 		})
 		It("should support well-known labels for a gpu (nvidia)", func() {
+			nodeClass.Spec.AMIFamily = lo.ToPtr(v1beta1.AMIFamilyAL2)
 			nodeSelector := map[string]string{
 				v1beta1.LabelInstanceGPUName:         "t4",
 				v1beta1.LabelInstanceGPUMemory:       "16384",