Skip to content

Commit 164ac37

Browse files
committed
tritony==0.0.14, remove too verbose log with tritonclient
1 parent c7c870a commit 164ac37

File tree

2 files changed

+5
-5
lines changed

2 files changed

+5
-5
lines changed

tritony/tools.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,7 @@
66
import logging
77
import os
88
import time
9+
import warnings
910
from concurrent.futures import ThreadPoolExecutor
1011
from typing import Any, Dict, List, Optional, Union
1112

@@ -19,10 +20,9 @@
1920
try:
2021
from tritonclient.grpc import _get_inference_request as grpc_get_inference_request
2122
except ImportError:
22-
logging.info("tritonclient[all]>=2.34.0")
23+
warnings.warn(UserWarning("tritonclient[all]>=2.34.0"))
2324
from tritonclient.grpc._utils import _get_inference_request as grpc_get_inference_request
2425

25-
2626
from tritonclient.utils import InferenceServerException
2727

2828
from tritony import ASYNC_TASKS
@@ -117,8 +117,8 @@ async def request_async(protocol: TritonProtocol, model_input: Dict, triton_clie
117117
if "parameters" in grpc_get_inference_request.__code__.co_varnames:
118118
# check tritonclient[all]>=2.34.0, NGC 23.04
119119
model_input["parameters"] = model_input.get("parameters", None)
120-
else:
121-
logger.warning("tritonclient[all]<2.34.0, NGC 21.04")
120+
elif "parameters" in model_input:
121+
warnings.warn(UserWarning("tritonclient[all]<2.34.0, NGC 23.04"))
122122
model_input.pop("parameters")
123123
request = grpc_get_inference_request(
124124
**model_input,

tritony/version.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
1-
__version__ = "0.0.12"
1+
__version__ = "0.0.14"

0 commit comments

Comments
 (0)