Fix jax E2E test (#38)

* Fix jax E2E test

Author: davidLif

hack/run-e2e-kind.sh

@@ -36,6 +36,6 @@ if [ ! -f ${GOBIN}/ginkgo ]; then
     GOBIN=${GOBIN} go install github.com/onsi/ginkgo/v2/[email protected]
 fi
 
-${GOBIN}/ginkgo -r --keep-going --randomize-all --randomize-suites --trace -vv ${REPO_ROOT}/test/e2e/suites/integrations/third_party --label-filter '!autoscale', '!scale'
+${GOBIN}/ginkgo -r --keep-going --randomize-all --randomize-suites --trace -vv ${REPO_ROOT}/test/e2e/suites --label-filter '!autoscale', '!scale'
 
 kind delete cluster --name $CLUSTER_NAME

test/e2e/suites/integrations/third_party/kubeflow.org/jax/jax_test.go

@@ -18,7 +18,7 @@ import (
 	runtimeClient "sigs.k8s.io/controller-runtime/pkg/client"
 
 	v2 "github.com/NVIDIA/KAI-scheduler/pkg/apis/scheduling/v2"
-	"github.com/NVIDIA/KAI-scheduler/pkg/scheduler/constants"
+	"github.com/NVIDIA/KAI-scheduler/pkg/common/constants"
 	testcontext "github.com/NVIDIA/KAI-scheduler/test/e2e/modules/context"
 	"github.com/NVIDIA/KAI-scheduler/test/e2e/modules/resources/capacity"
 	"github.com/NVIDIA/KAI-scheduler/test/e2e/modules/resources/rd"
@@ -65,10 +65,10 @@ var _ = Describe("JAX integration", Ordered, func() {
 	It("should run the pods of the Jax", func(ctx context.Context) {
 		singleGPURequest := v1.ResourceRequirements{
 			Limits: v1.ResourceList{
-				constants.GPUResource: resource.MustParse("1"),
+				constants.GpuResource: resource.MustParse("1"),
 			},
 			Requests: v1.ResourceList{
-				constants.GPUResource: resource.MustParse("1"),
+				constants.GpuResource: resource.MustParse("1"),
 			},
 		}