Support tritonclient>=2.34.0, Loosely modified the requirements related to tritonclient

Author: kimdwkimdw

README.md

@@ -53,6 +53,12 @@ if __name__ == "__main__":
         print(f"{max_value} ({arg_max}) = {class_name}")
 ```
 
+## Release Notes
+
+- 23.06.16 Support tritonclient>=2.34.0
+- Loosely modified the requirements related to tritonclient
+
+
 ## Key Features
 
 - [x] Simple configuration. Only `$host:$port` and `$model_name` are required.

setup.cfg

@@ -44,8 +44,8 @@ zip_safe = False
 include_package_data = True
 packages = find:
 install_requires =
-    tritonclient[all]>=2.18.0
-    protobuf>=3.5.0,<3.20
+    tritonclient[all]>=2.21.0
+    protobuf>=3.5.0
     orjson>=3.6.8
     reretry>=0.11.1
     attrs>=21.2.0
@@ -56,6 +56,9 @@ exclude =
     tools*
     docs*
 
+# https://github.com/triton-inference-server/client/blob/main/src/python/library/setup.py#L118
+# official package is supporting python 3.6+, but python 3.6, 3.7 is not supported.
+# https://devguide.python.org/versions/
 python_requires = >=3.8
 
 

tritony/tools.py

@@ -16,7 +16,14 @@
 from tritonclient import grpc as grpcclient
 from tritonclient import http as httpclient
 from tritonclient.grpc import InferResult
-from tritonclient.grpc import _get_inference_request as grpc_get_inference_request
+
+try:
+    from tritonclient.grpc import _get_inference_request as grpc_get_inference_request
+except ImportError:
+    logging.info("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

tritony/version.py

@@ -1 +1 @@
-__version__ = "0.0.10"
+__version__ = "0.0.11"