From 937da8b90379a10fc10113140d6180f27c36d527 Mon Sep 17 00:00:00 2001 From: fpetrini15 Date: Wed, 3 Jan 2024 13:38:18 -0800 Subject: [PATCH] Initial commit --- 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 1ea8450d2..ce00c107f 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.'