Add additional testing

triton-inference-server · Jan 22, 2024 · 0fc3acc · 0fc3acc
1 parent 953e7fe
commit 0fc3acc
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/qa/L0_http/generate_endpoint_test.py b/qa/L0_http/generate_endpoint_test.py
@@ -343,7 +343,7 @@ def test_complex_schema(self):
         inputs = {
             "PROMPT": "hello world",
             "STREAM": True,
-            "PARAMS": {"PARAM_0": 0, "PARAM_1": True},
+            "PARAMS": {"PARAM_0": 0, "PARAM_1": True, "PARAM_2": 123.123},
         }
         r = self.generate(self._model_name, inputs)
         try:

diff --git a/src/http_server.cc b/src/http_server.cc
@@ -3822,7 +3822,7 @@ HTTPAPIServer::InferRequestClass::FinalizeResponse(
               name, reinterpret_cast<const char*>(vvalue)));
           break;
         case TRITONSERVER_PARAMETER_DOUBLE:
-          RETURN_IF_ERR(params_json.AddInt(
+          RETURN_IF_ERR(params_json.AddDouble(
               name, *(reinterpret_cast<const double*>(vvalue))));
           break;
         case TRITONSERVER_PARAMETER_BYTES: