From 6f4b7a95feb2c0a89b4534428cfdaa14e26fdca7 Mon Sep 17 00:00:00 2001 From: Sivanantham <90966311+sivanantha321@users.noreply.github.com> Date: Sun, 4 Feb 2024 00:20:34 +0530 Subject: [PATCH] Fix v2 model ready url in kserve client (#3403) Signed-off-by: Sivanantham Chinnaiyan --- python/kserve/kserve/api/kserve_client.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/python/kserve/kserve/api/kserve_client.py b/python/kserve/kserve/api/kserve_client.py index 82fb2863eeb..8fe11069fca 100644 --- a/python/kserve/kserve/api/kserve_client.py +++ b/python/kserve/kserve/api/kserve_client.py @@ -422,6 +422,8 @@ def wait_model_ready(self, service_name, model_name, isvc_namespace=None, # pyl time.sleep(polling_interval) # Check model health API url = f"http://{cluster_ip}/{protocol_version}/models/{model_name}" + if protocol_version.lower() == "v2": + url = f"http://{cluster_ip}/{protocol_version}/models/{model_name}/ready" response = requests.get(url, headers=headers).status_code if response == 200: return