test(huggingface): Use correct content type

I'm not sure how this was passing before, as it was wrong [1][2], but it's fixed now. [1] https://github.com/SeldonIO/MLServer/blob/b009cd0eb1dde7dc2da440dfdf9dd091c4c6b5d0/runtimes/huggingface/mlserver_huggingface/codecs/base.py#L197 [2] https://github.com/SeldonIO/MLServer/blob/695affb2758bc3f57230ccf37210a6e5a507a040/runtimes/huggingface/mlserver_huggingface/codecs/jsonlist.py#L15
SeldonIO · May 10, 2024 · 3b7d222 · 3b7d222
1 parent b009cd0
commit 3b7d222
Showing 1 changed file with 9 additions and 9 deletions.
diff --git a/runtimes/huggingface/tests/test_codecs.py b/runtimes/huggingface/tests/test_codecs.py
@@ -44,7 +44,7 @@ def test_decode_request(inference_request, expected):
             {"foo": ["bar1", "bar2"], "foo2": ["var1"]},
             True,
             InferenceRequest(
-                parameters=Parameters(content_type="str"),
+                parameters=Parameters(content_type="hf"),
                 inputs=[
                     RequestInput(
                         name="foo",
@@ -68,7 +68,7 @@ def test_decode_request(inference_request, expected):
             False,
             InferenceRequest(
                 model_name="my-model",
-                parameters=Parameters(content_type="str"),
+                parameters=Parameters(content_type="hf"),
                 inputs=[
                     RequestInput(
                         name="foo",
@@ -135,10 +135,10 @@ def test_decode_response(inference_response, expected):
                 model_name="my-model",
                 outputs=[
                     ResponseOutput(
-                        name="output_0",
-                        shape=[1],
+                        name="output",
+                        shape=[1, 1],
                         datatype="BYTES",
-                        parameters=Parameters(content_type="hg_json"),
+                        parameters=Parameters(content_type="hg_jsonlist"),
                         data=[b'{"foo": ["bar1", "bar2"], "foo2": ["var1"]}'],
                     )
                 ],
@@ -151,11 +151,11 @@ def test_decode_response(inference_response, expected):
                 model_name="my-model",
                 outputs=[
                     ResponseOutput(
-                        name="output_0",
-                        shape=[1],
+                        name="output",
+                        shape=[1, 1],
                         datatype="BYTES",
-                        parameters=Parameters(content_type="hg_json"),
-                        data=['{"foo": ["bar1", "bar2"], "foo2": ["var1"]}'],
+                        parameters=Parameters(content_type="hg_jsonlist"),
+                        data=[b'{"foo": ["bar1", "bar2"], "foo2": ["var1"]}'],
                     )
                 ],
             ),