Initial commit

triton-inference-server · Jan 3, 2024 · 937da8b · 937da8b
1 parent c21fe5a
commit 937da8b
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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.'