From 0fa5f3ce2489a0375eeda98b3d595480ff8ac554 Mon Sep 17 00:00:00 2001
From: Shilpa Chugh <shchugh@redhat.com>
Date: Tue, 19 Nov 2024 14:13:57 +0530
Subject: [PATCH] Hpo: Remove deprecated parameters

---
 tests/odh/resources/mnist_hpo_raytune.ipynb | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/tests/odh/resources/mnist_hpo_raytune.ipynb b/tests/odh/resources/mnist_hpo_raytune.ipynb
index 4d8c2343..6f8715bc 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",