diff --git a/tests/integration/ray_test.go b/tests/integration/ray_test.go index 365b0b39..9f309d83 100644 --- a/tests/integration/ray_test.go +++ b/tests/integration/ray_test.go @@ -276,12 +276,12 @@ func createRayCluster(test support.Test, namespace string, mnist *corev1.ConfigM }, Resources: corev1.ResourceRequirements{ Requests: corev1.ResourceList{ - corev1.ResourceCPU: resource.MustParse("250m"), - corev1.ResourceMemory: resource.MustParse("512Mi"), + corev1.ResourceCPU: resource.MustParse("2"), + corev1.ResourceMemory: resource.MustParse("8G"), }, Limits: corev1.ResourceList{ - corev1.ResourceCPU: resource.MustParse("1"), - corev1.ResourceMemory: resource.MustParse("1G"), + corev1.ResourceCPU: resource.MustParse("2"), + corev1.ResourceMemory: resource.MustParse("8G"), }, }, VolumeMounts: []corev1.VolumeMount{ @@ -336,12 +336,12 @@ func createRayCluster(test support.Test, namespace string, mnist *corev1.ConfigM }, Resources: corev1.ResourceRequirements{ Requests: corev1.ResourceList{ - corev1.ResourceCPU: resource.MustParse("250m"), - corev1.ResourceMemory: resource.MustParse("256Mi"), + corev1.ResourceCPU: resource.MustParse("500m"), + corev1.ResourceMemory: resource.MustParse("512Mi"), }, Limits: corev1.ResourceList{ - corev1.ResourceCPU: resource.MustParse("1"), - corev1.ResourceMemory: resource.MustParse("512Mi"), + corev1.ResourceCPU: resource.MustParse("2"), + corev1.ResourceMemory: resource.MustParse("1Gi"), }, }, }, diff --git a/tests/integration/resources/mnist_ray_mini.ipynb b/tests/integration/resources/mnist_ray_mini.ipynb index eca44f4a..a5b61153 100644 --- a/tests/integration/resources/mnist_ray_mini.ipynb +++ b/tests/integration/resources/mnist_ray_mini.ipynb @@ -40,7 +40,7 @@ "outputs": [], "source": [ "# Create our cluster and submit appwrapper\n", - "cluster = Cluster(ClusterConfiguration(namespace=namespace, name='mnisttest', num_workers=4, min_cpus=4, max_cpus=4, min_memory=8, max_memory=8, num_gpus=0, instascale=False))" + "cluster = Cluster(ClusterConfiguration(namespace=namespace, name='mnisttest', head_cpus=2, head_memory=8, num_workers=1, min_cpus=1, max_cpus=1, min_memory=4, max_memory=4, num_gpus=0, instascale=False))" ] }, {