From 82cf0cceedc8a6cdd2907af6c3a569c814f5fe38 Mon Sep 17 00:00:00 2001 From: Krishnan Prashanth Date: Fri, 29 Nov 2024 16:09:52 -0800 Subject: [PATCH] Raises error for http health checks --- src/python/library/tritonclient/http/_client.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/src/python/library/tritonclient/http/_client.py b/src/python/library/tritonclient/http/_client.py index 1b71d82a7..6c0c91684 100755 --- a/src/python/library/tritonclient/http/_client.py +++ b/src/python/library/tritonclient/http/_client.py @@ -366,6 +366,8 @@ def is_server_live(self, headers=None, query_params=None): request_uri=request_uri, headers=headers, query_params=query_params ) + _raise_if_error(response) + return response.status_code == 200 def is_server_ready(self, headers=None, query_params=None): @@ -396,6 +398,8 @@ def is_server_ready(self, headers=None, query_params=None): request_uri=request_uri, headers=headers, query_params=query_params ) + _raise_if_error(response) + return response.status_code == 200 def is_model_ready(