Changed to regular ubuntu docker image in tests (#123)

Author: romanbaron

hack/run-e2e-kind.sh

@@ -18,13 +18,8 @@ fi
 
 kind create cluster --config ${KIND_CONFIG} --name $CLUSTER_NAME
 
-# Add necessary helm repos
-helm repo add fake-gpu https://runai.jfrog.io/artifactory/api/helm/fake-gpu-operator-charts-prod
-helm repo update
-
 # Install the fake-gpu-operator to provide a fake GPU resources for the e2e tests
-helm upgrade -i gpu-operator fake-gpu/fake-gpu-operator --namespace gpu-operator --create-namespace --version 0.0.53 --set topology.nodePools.default.gpuCount=8
-sleep 10 # Wait for the fake-gpu-operator to start
+helm upgrade -i gpu-operator oci://ghcr.io/run-ai/fake-gpu-operator/fake-gpu-operator --namespace gpu-operator --create-namespace --version 0.0.62 --set topology.nodePools.default.gpuCount=8 --wait
 
 # install third party operators to check the compatibility with the kai-scheduler
 if [ "$TEST_THIRD_PARTY_INTEGRATIONS" = "true" ]; then

pkg/env-tests/utils.go

@@ -101,7 +101,7 @@ func CreatePodObject(namespace, name string, resources corev1.ResourceRequiremen
 		Spec: corev1.PodSpec{
 			Containers: []corev1.Container{
 				{
-					Image: "gcr.io/run-ai-lab/ubuntu",
+					Image: "ubuntu",
 					Name:  "ubuntu-container",
 					Args: []string{
 						"sleep",

test/e2e/modules/resources/rd/cronjob.go

@@ -51,7 +51,7 @@ func CreateCronJobObject(namespace, queueName string) *batchv1.CronJob {
 							TerminationGracePeriodSeconds: pointer.Int64(0),
 							Containers: []v1.Container{
 								{
-									Image: "gcr.io/run-ai-lab/ubuntu",
+									Image: "ubuntu",
 									Name:  "ubuntu-container",
 									Args: []string{
 										"sleep",

test/e2e/modules/resources/rd/deployment.go

@@ -48,7 +48,7 @@ func CreateDeploymentObject(namespace, queueName string) *appsv1.Deployment {
 				Spec: v1.PodSpec{
 					Containers: []v1.Container{
 						{
-							Image: "gcr.io/run-ai-lab/ubuntu",
+							Image: "ubuntu",
 							Name:  "ubuntu-container",
 							Args: []string{
 								"sleep",

test/e2e/modules/resources/rd/mpi/mpi.go

@@ -89,7 +89,7 @@ func getPodTemplate(queueName, matchLabelValue string) corev1.PodTemplateSpec {
 			TerminationGracePeriodSeconds: pointer.Int64(0),
 			Containers: []corev1.Container{
 				{
-					Image: "gcr.io/run-ai-lab/ubuntu",
+					Image: "ubuntu",
 					Name:  "ubuntu-container",
 					Args: []string{
 						"sleep",

test/e2e/modules/resources/rd/pod.go

@@ -146,7 +146,7 @@ func CreatePodObject(podQueue *v2.Queue, resources v1.ResourceRequirements) *v1.
 		Spec: v1.PodSpec{
 			Containers: []v1.Container{
 				{
-					Image: "nvcr.io/nvidia/base/ubuntu:22.04_20240212",
+					Image: "ubuntu",
 					Name:  "ubuntu-container",
 					Args: []string{
 						"sleep",

test/e2e/modules/resources/rd/pytorch/pytorch.go

@@ -82,7 +82,7 @@ func GetPodTemplate(queueName, matchLabelValue string) corev1.PodTemplateSpec {
 			TerminationGracePeriodSeconds: pointer.Int64(0),
 			Containers: []corev1.Container{
 				{
-					Image: "gcr.io/run-ai-lab/ubuntu",
+					Image: "ubuntu",
 					Name:  "pytorch",
 					Args: []string{
 						"sleep",

test/e2e/modules/resources/rd/replicaset.go

@@ -48,7 +48,7 @@ func CreateReplicasetObject(namespace, queueName string) *v1.ReplicaSet {
 				Spec: v12.PodSpec{
 					Containers: []v12.Container{
 						{
-							Image: "gcr.io/run-ai-lab/ubuntu",
+							Image: "ubuntu",
 							Name:  "ubuntu-container",
 							Args: []string{
 								"sleep",

test/e2e/modules/resources/rd/statefulset.go

@@ -50,7 +50,7 @@ func CreateStatefulSetObject(namespace, queueName string) *v1.StatefulSet {
 				Spec: v12.PodSpec{
 					Containers: []v12.Container{
 						{
-							Image: "gcr.io/run-ai-lab/ubuntu",
+							Image: "ubuntu",
 							Name:  "ubuntu-container",
 							Args: []string{
 								"sleep",