diff --git a/tests/odh/resources/mnist_hpo_raytune.ipynb b/tests/odh/resources/mnist_hpo_raytune.ipynb index 6179d213..ae6f067b 100644 --- a/tests/odh/resources/mnist_hpo_raytune.ipynb +++ b/tests/odh/resources/mnist_hpo_raytune.ipynb @@ -76,8 +76,10 @@ " ClusterConfiguration(\n", " namespace=namespace,\n", " name='mnisthpotest',\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",