Add ONNX model format testcase in MLServer test-suite

Signed-off-by: Snomaan6846 <syedali@redhat.com>

rh-pre-commit.version: 2.3.2
rh-pre-commit.check-secrets: ENABLED

Author: Snomaan6846

tests/model_serving/model_runtime/mlserver/basic_model_deployment/__snapshots__/test_mlserver_basic_model_deployment/TestMLServerModels.test_mlserver_model_inference[onnx-raw-deployment].json

@@ -0,0 +1,22 @@
+{
+  "id": "onnx",
+  "model_name": "onnx",
+  "model_version": "v1.0.0",
+  "outputs": [
+    {
+      "data": [
+        -137.39060974121094
+      ],
+      "datatype": "FP32",
+      "name": "predict",
+      "parameters": {
+        "content_type": "np"
+      },
+      "shape": [
+        1,
+        1
+      ]
+    }
+  ],
+  "parameters": {}
+}

tests/model_serving/model_runtime/mlserver/basic_model_deployment/test_mlserver_basic_model_deployment.py

@@ -51,6 +51,20 @@
                 deployment_type=RAW_DEPLOYMENT_TYPE,
             ),
         ),
+        pytest.param(
+            get_model_namespace_dict(
+                model_format_name=ModelFormat.ONNX,
+                deployment_type=RAW_DEPLOYMENT_TYPE,
+            ),
+            get_deployment_config_dict(model_format_name=ModelFormat.ONNX, deployment_type=RAW_DEPLOYMENT_TYPE),
+            get_model_storage_uri_dict(model_format_name=ModelFormat.ONNX),
+            get_deployment_config_dict(model_format_name=ModelFormat.ONNX, deployment_type=RAW_DEPLOYMENT_TYPE),
+            ModelFormat.ONNX,
+            id=get_test_case_id(
+                model_format_name=ModelFormat.ONNX,
+                deployment_type=RAW_DEPLOYMENT_TYPE,
+            ),
+        ),
         pytest.param(
             get_model_namespace_dict(
                 model_format_name=ModelFormat.SKLEARN,

tests/model_serving/model_runtime/mlserver/constant.py

@@ -77,13 +77,31 @@ class OutputType:
     ],
 }
 
+ONNX_REST_INPUT_QUERY = {
+    "id": "onnx",
+    "inputs": [
+        {
+            "name": "input",
+            "shape": [1, 4],
+            "datatype": "FP32",
+            "data": [[-1.44964521969853, -0.6500239344068982, -0.08343796979036086, -1.496529255090079]],
+        }
+    ],
+}
+
 MODEL_CONFIGS: dict[str, dict[str, Any]] = {
     ModelFormat.LIGHTGBM: {
         "model_name": ModelFormat.LIGHTGBM,
         "model_version": "v0.1.0",
         "rest_query": LIGHTGBM_REST_INPUT_QUERY,
         "output_type": OutputType.DETERMINISTIC,
     },
+    ModelFormat.ONNX: {
+        "model_name": ModelFormat.ONNX,
+        "model_version": "v1.0.0",
+        "rest_query": ONNX_REST_INPUT_QUERY,
+        "output_type": OutputType.DETERMINISTIC,
+    },
     ModelFormat.SKLEARN: {
         "model_name": ModelFormat.SKLEARN,
         "model_version": "v1.0.0",