diff --git a/tests/odh/resources/mnist_ray_mini.ipynb b/tests/odh/resources/mnist_ray_mini.ipynb index fbabd227..44235225 100644 --- a/tests/odh/resources/mnist_ray_mini.ipynb +++ b/tests/odh/resources/mnist_ray_mini.ipynb @@ -76,8 +76,10 @@ " ClusterConfiguration(\n", " namespace=namespace,\n", " name='mnisttest',\n", - " head_cpus=1,\n", - " head_memory=4,\n", + " head_cpu_requests=1,\n", + " head_cpu_limits=1,\n", + " head_memory_requests=4,\n", + " head_memory_limits=4,\n", " head_extended_resource_requests={'nvidia.com/gpu':0},\n", " num_workers=1,\n", " worker_cpu_requests=1,\n",