[KAI integration] Fixing tests

EkinKarabulut · EkinKarabulut · commit 1a63219ee327 · 2025-07-30T09:27:53.000+02:00
diff --git a/ray-operator/controllers/ray/batchscheduler/kai-scheduler/kai_scheduler.go b/ray-operator/controllers/ray/batchscheduler/kai-scheduler/kai_scheduler.go
@@ -18,6 +18,7 @@ import (
 
 	rayv1 "github.com/ray-project/kuberay/ray-operator/apis/ray/v1"
 	schedulerinterface "github.com/ray-project/kuberay/ray-operator/controllers/ray/batchscheduler/interface"
+	"sigs.k8s.io/controller-runtime/pkg/client"
 )
 
 const (
@@ -53,7 +54,7 @@ func (k *KaiScheduler) AddMetadataToPod(ctx context.Context, app *rayv1.RayClust
 	pod.Labels[QueueLabelName] = queue
 }
 
-func (kf *KaiSchedulerFactory) New(_ context.Context, _ *rest.Config) (schedulerinterface.BatchScheduler, error) {
+func (kf *KaiSchedulerFactory) New(_ context.Context, _ *rest.Config, _ client.Client) (schedulerinterface.BatchScheduler, error) {
 	return &KaiScheduler{}, nil
 }
 

Original file line number	Diff line number	Diff line change
`@@ -18,6 +18,7 @@ import (`
`18`	`18`
`19`	`19`	`rayv1 "github.com/ray-project/kuberay/ray-operator/apis/ray/v1"`
`20`	`20`	`schedulerinterface "github.com/ray-project/kuberay/ray-operator/controllers/ray/batchscheduler/interface"`
	`21`	`+ "sigs.k8s.io/controller-runtime/pkg/client"`
`21`	`22`	`)`
`22`	`23`
`23`	`24`	`const (`
`@@ -53,7 +54,7 @@ func (k KaiScheduler) AddMetadataToPod(ctx context.Context, app rayv1.RayClust`
`53`	`54`	`pod.Labels[QueueLabelName] = queue`
`54`	`55`	`}`
`55`	`56`
`56`		`-func (kf KaiSchedulerFactory) New(_ context.Context, _ rest.Config) (schedulerinterface.BatchScheduler, error) {`
	`57`	`+func (kf KaiSchedulerFactory) New(_ context.Context, _ rest.Config, _ client.Client) (schedulerinterface.BatchScheduler, error) {`
`57`	`58`	`return &KaiScheduler{}, nil`
`58`	`59`	`}`
`59`	`60`