From 5edd3b5d90d3e16cdfd9e31281e316fa8883adb4 Mon Sep 17 00:00:00 2001 From: SiorMeir Date: Thu, 5 Mar 2026 13:10:32 +0200 Subject: [PATCH] fix: port kwok_test fix --- test/e2e/scale/kwok_test.go | 2 ++ 1 file changed, 2 insertions(+) diff --git a/test/e2e/scale/kwok_test.go b/test/e2e/scale/kwok_test.go index 736e0b3ff..3219c3aa8 100644 --- a/test/e2e/scale/kwok_test.go +++ b/test/e2e/scale/kwok_test.go @@ -370,6 +370,7 @@ var _ = Describe("Kwok scale test", Ordered, Label(labels.Scale), func() { Context("measure reclaim failure time", func() { BeforeAll(func(ctx context.Context) { + Expect(testCtx.ControllerClient.Get(ctx, runtimeClient.ObjectKeyFromObject(sanityTestQueue), sanityTestQueue)).To(Succeed()) sanityTestQueue.Spec.Resources.GPU.Quota = float64((numberOfNodes * gpusPerNode) - (defaultPodsPerDistributedJob * gpusPerNode) + 1) Expect(testCtx.ControllerClient.Update(ctx, sanityTestQueue)).To(Succeed()) }) @@ -378,6 +379,7 @@ var _ = Describe("Kwok scale test", Ordered, Label(labels.Scale), func() { if CurrentSpecReport().Failed() { return } + Expect(testCtx.ControllerClient.Get(ctx, runtimeClient.ObjectKeyFromObject(sanityTestQueue), sanityTestQueue)).To(Succeed()) sanityTestQueue.Spec.Resources.GPU.Quota = 0 Expect(testCtx.ControllerClient.Update(ctx, sanityTestQueue)).To(Succeed()) })