Put timeout in RPC call, not protobuf object (#436)

triton-inference-server · Nov 21, 2023 · 2a125ee · 2a125ee
1 parent 80c1d89
commit 2a125ee
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/python/library/tritonclient/grpc/aio/__init__.py b/src/python/library/tritonclient/grpc/aio/__init__.py
@@ -357,7 +357,7 @@ async def get_inference_statistics(
             if type(model_version) != str:
                 raise_error("model version must be a string")
             request = service_pb2.ModelStatisticsRequest(
-                name=model_name, version=model_version, timeout=client_timeout
+                name=model_name, version=model_version
             )
             if self._verbose:
                 print(
@@ -366,7 +366,7 @@ async def get_inference_statistics(
                     )
                 )
             response = await self._client_stub.ModelStatistics(
-                request=request, metadata=metadata
+                request=request, metadata=metadata, timeout=client_timeout
             )
             if self._verbose:
                 print(response)