Fix shape

SeldonIO · Oct 19, 2020 · aad1ab6 · aad1ab6
1 parent e04b110
commit aad1ab6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/testing/scripts/test_prepackaged_servers.py b/testing/scripts/test_prepackaged_servers.py
@@ -64,7 +64,7 @@ def test_sklearn_v2(self, namespace):
                 "inputs": [
                     {
                         "name": "input-0",
-                        "shape": [4],
+                        "shape": [1, 4],
                         "datatype": "FP32",
                         "data": [[0.1, 0.2, 0.3, 0.4]],
                     }
@@ -133,7 +133,7 @@ def test_xgboost_v2(self, namespace):
                 "inputs": [
                     {
                         "name": "input-0",
-                        "shape": [4],
+                        "shape": [1, 4],
                         "datatype": "FP32",
                         "data": [[0.1, 0.2, 0.3, 0.4]],
                     }