Fix v2 model ready url in kserve client (kserve#3403)

Signed-off-by: Sivanantham Chinnaiyan <[email protected]>
sivanantha321 · Feb 3, 2024 · 6f4b7a9 · 6f4b7a9
1 parent ad88ab4
commit 6f4b7a9
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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