From f3d25310cc1bc6b276ebf4e77655b408c0fa1bcc Mon Sep 17 00:00:00 2001 From: "jun.dai" Date: Fri, 26 Apr 2024 14:13:17 +0800 Subject: [PATCH] [fix] return the infer_response while response.error_message is empty to associate it with request. --- src/python/library/tritonclient/grpc/_infer_stream.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/src/python/library/tritonclient/grpc/_infer_stream.py b/src/python/library/tritonclient/grpc/_infer_stream.py index 92ec05890..947a31426 100755 --- a/src/python/library/tritonclient/grpc/_infer_stream.py +++ b/src/python/library/tritonclient/grpc/_infer_stream.py @@ -151,8 +151,7 @@ def _process_response(self): result = error = None if response.error_message != "": error = InferenceServerException(msg=response.error_message) - else: - result = InferResult(response.infer_response) + result = InferResult(response.infer_response) self._callback(result=result, error=error) except grpc.RpcError as rpc_error: # On GRPC error, refresh the active state to indicate if the stream