From 36584f32f3c44ced0b28bb3ba0447ed2235b032e Mon Sep 17 00:00:00 2001 From: nikshorop Date: Fri, 28 Mar 2025 13:38:29 +0300 Subject: [PATCH] fix get_output while as_json=True --- src/python/library/tritonclient/grpc/_infer_result.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/python/library/tritonclient/grpc/_infer_result.py b/src/python/library/tritonclient/grpc/_infer_result.py index ff3275b02..d5f8dbccb 100755 --- a/src/python/library/tritonclient/grpc/_infer_result.py +++ b/src/python/library/tritonclient/grpc/_infer_result.py @@ -124,7 +124,9 @@ def get_output(self, name, as_json=False): for output in self._result.outputs: if output.name == name: if as_json: - MessageToJson(output, preserving_proto_field_name=True) + return json.loads( + MessageToJson(output, preserving_proto_field_name=True) + ) else: return output