diff --git a/tests/model_serving/model_server/utils.py b/tests/model_serving/model_server/utils.py index 6fcfb9ae..892750e8 100644 --- a/tests/model_serving/model_server/utils.py +++ b/tests/model_serving/model_server/utils.py @@ -14,11 +14,11 @@ from tenacity import retry, wait_exponential, stop_after_attempt from utilities.constants import KServeDeploymentType from utilities.inference_utils import UserInference -from utilities.infra import ( - get_pods_by_isvc_label, - wait_for_inference_deployment_replicas, -) +from utilities.infra import wait_for_inference_deployment_replicas + LOGGER = get_logger(name=__name__) + + class InferenceResponseError(Exception): pass