Initial commit (#453)

* Support Double-Type Infer/Response Parameters
triton-inference-server · Feb 1, 2024 · 1dc9836 · 1dc9836
1 parent 24c1ff7
commit 1dc9836
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.'