Resource request and limits

Author: ChristianZaccaria

tests/e2e/mnist_raycluster_sdk.py

@@ -9,17 +9,19 @@
 
 namespace = sys.argv[1]
 
-cluster = Cluster(ClusterConfiguration(
-    name='mnist',
-    namespace=namespace,
-    num_workers=1,
-    min_cpus='500m',
-    max_cpus=1,
-    min_memory=0.5,
-    max_memory=1,
-    num_gpus=0,
-    instascale=False,
-))
+cluster = Cluster(
+    ClusterConfiguration(
+        name="mnist",
+        namespace=namespace,
+        num_workers=1,
+        min_cpus="500m",
+        max_cpus=1,
+        min_memory=0.5,
+        max_memory=1,
+        num_gpus=0,
+        instascale=False,
+    )
+)
 
 cluster.up()
 

tests/e2e/mnist_raycluster_sdk_test.go

@@ -19,6 +19,8 @@ package e2e
 import (
 	"testing"
 
+	"k8s.io/apimachinery/pkg/api/resource"
+
 	. "github.com/onsi/gomega"
 	mcadv1beta1 "github.com/project-codeflare/multi-cluster-app-dispatcher/pkg/apis/controller/v1beta1"
 	rayv1alpha1 "github.com/ray-project/kuberay/ray-operator/apis/ray/v1alpha1"
@@ -174,6 +176,16 @@ func TestMNISTRayClusterSDK(t *testing.T) {
 									MountPath: "/test",
 								},
 							},
+							Resources: corev1.ResourceRequirements{
+								Requests: corev1.ResourceList{
+									corev1.ResourceCPU:    resource.MustParse("250m"),
+									corev1.ResourceMemory: resource.MustParse("512Mi"),
+								},
+								Limits: corev1.ResourceList{
+									corev1.ResourceCPU:    resource.MustParse("500m"),
+									corev1.ResourceMemory: resource.MustParse("1G"),
+								},
+							},
 						},
 					},
 					Volumes: []corev1.Volume{