diff --git a/internal/controller/gpupool_controller_test.go b/internal/controller/gpupool_controller_test.go index 52a2a9d..bfbfd72 100644 --- a/internal/controller/gpupool_controller_test.go +++ b/internal/controller/gpupool_controller_test.go @@ -130,7 +130,8 @@ var _ = Describe("GPUPool Controller", func() { Build() updateRollingUpdatePolicy(tfEnv, true, 50, "3s") newHash, _ := triggerHypervisorUpdate(tfEnv) - verifyAllHypervisorPodHash(tfEnv, newHash) + verifyHypervisorPodHash(tfEnv.GetGPUNode(0, 0), newHash) + verifyHypervisorPodHash(tfEnv.GetGPUNode(0, 1), newHash) verifyHypervisorUpdateProgress(tfEnv, 100) tfEnv.Cleanup() })