From 1dc9836f4a4a8d4757ecf6bdfea9635624ce862b Mon Sep 17 00:00:00 2001 From: Francesco Petrini Date: Thu, 1 Feb 2024 10:06:43 -0800 Subject: [PATCH] Initial commit (#453) * Support Double-Type Infer/Response Parameters --- src/python/library/tritonclient/grpc/_utils.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/src/python/library/tritonclient/grpc/_utils.py b/src/python/library/tritonclient/grpc/_utils.py index 8dd48f785..dae6d71f8 100755 --- a/src/python/library/tritonclient/grpc/_utils.py +++ b/src/python/library/tritonclient/grpc/_utils.py @@ -133,6 +133,8 @@ def _get_inference_request( request.parameters[key].bool_param = value elif isinstance(value, int): request.parameters[key].int64_param = value + elif isinstance(value, float): + request.parameters[key].double_param = value else: raise_error( f'The parameter datatype "{type(value)}" for key "{key}" is not supported.'