tritony==0.0.14, remove too verbose log with tritonclient

rtzr · Sep 13, 2023 · 164ac37 · 164ac37
1 parent c7c870a
commit 164ac37
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/tritony/tools.py b/tritony/tools.py
@@ -6,6 +6,7 @@
 import logging
 import os
 import time
+import warnings
 from concurrent.futures import ThreadPoolExecutor
 from typing import Any, Dict, List, Optional, Union
 
@@ -19,10 +20,9 @@
 try:
     from tritonclient.grpc import _get_inference_request as grpc_get_inference_request
 except ImportError:
-    logging.info("tritonclient[all]>=2.34.0")
+    warnings.warn(UserWarning("tritonclient[all]>=2.34.0"))
     from tritonclient.grpc._utils import _get_inference_request as grpc_get_inference_request
 
-
 from tritonclient.utils import InferenceServerException
 
 from tritony import ASYNC_TASKS
@@ -117,8 +117,8 @@ async def request_async(protocol: TritonProtocol, model_input: Dict, triton_clie
         if "parameters" in grpc_get_inference_request.__code__.co_varnames:
             # check tritonclient[all]>=2.34.0, NGC 23.04
             model_input["parameters"] = model_input.get("parameters", None)
-        else:
-            logger.warning("tritonclient[all]<2.34.0, NGC 21.04")
+        elif "parameters" in model_input:
+            warnings.warn(UserWarning("tritonclient[all]<2.34.0, NGC 23.04"))
             model_input.pop("parameters")
         request = grpc_get_inference_request(
             **model_input,

diff --git a/tritony/version.py b/tritony/version.py
@@ -1 +1 @@
-__version__ = "0.0.12"
+__version__ = "0.0.14"