diff --git a/pkg/controller/setup.go b/pkg/controller/setup.go index 9e0cfcc0..c3cae2d2 100644 --- a/pkg/controller/setup.go +++ b/pkg/controller/setup.go @@ -33,6 +33,7 @@ import ( "github.com/project-codeflare/appwrapper/internal/webhook" "github.com/project-codeflare/appwrapper/pkg/config" + "sigs.k8s.io/kueue/apis/config/v1beta1" "sigs.k8s.io/kueue/pkg/controller/jobframework" ) @@ -43,6 +44,7 @@ func SetupControllers(mgr ctrl.Manager, awConfig *config.AppWrapperConfig) error mgr.GetClient(), mgr.GetEventRecorderFor("kueue"), jobframework.WithManageJobsWithoutQueueName(true), + jobframework.WithWaitForPodsReady(&v1beta1.WaitForPodsReady{Enable: true}), ).SetupWithManager(mgr); err != nil { return fmt.Errorf("workload controller: %w", err) }