fix: pod group controller fails on missing priority class (#271)

Author: enoodle

pkg/podgroupcontroller/utilities/pod-group/preemptible.go

@@ -7,8 +7,10 @@ import (
 	"context"
 
 	schedulingv1 "k8s.io/api/scheduling/v1"
+	"k8s.io/apimachinery/pkg/api/errors"
 	"k8s.io/apimachinery/pkg/types"
 	"sigs.k8s.io/controller-runtime/pkg/client"
+	"sigs.k8s.io/controller-runtime/pkg/log"
 
 	"github.com/NVIDIA/KAI-scheduler/pkg/apis/scheduling/v2alpha2"
 )
@@ -19,6 +21,16 @@ const (
 
 func IsPreemptible(ctx context.Context, podGroup *v2alpha2.PodGroup, kubeClient client.Client) (bool, error) {
 	priority, err := getPodGroupPriority(ctx, podGroup, kubeClient)
+	if errors.IsNotFound(err) {
+		logger := log.FromContext(ctx)
+		logger.Info(
+			"Priority Class not found",
+			"podGroup", podGroup.Name,
+			"namespace", podGroup.Namespace,
+			"priorityClass", podGroup.Spec.PriorityClassName,
+		)
+		return true, nil
+	}
 	if err != nil {
 		return false, err
 	}

pkg/podgroupcontroller/utilities/pod-group/preemptible_test.go

@@ -110,8 +110,8 @@ func TestIsPreemptibleJob(t *testing.T) {
 					},
 				},
 			},
-			false,
 			true,
+			false,
 		},
 		{
 			"Custom preemptable class",