Fix image retrieval test to use data -> payload format

triton-inference-server · Nov 13, 2024 · eff74c8 · eff74c8
1 parent 7720ecd
commit eff74c8
Showing 1 changed file with 17 additions and 9 deletions.
diff --git a/genai-perf/tests/test_image_retrieval_converter.py b/genai-perf/tests/test_image_retrieval_converter.py
@@ -86,14 +86,22 @@ def test_convert_multi_modal_batched(self) -> None:
         result = image_retrieval_converter.convert(generic_dataset, config)
 
         expected_result = {
-            "input": [{
-                "type": "image_url",
-                "url": "test_image_1"
-            },
-            {
-                "type": "image_url",
-                "url": "test_image_2"
-            }]
+            "data": [
+                {
+                    "payload": [
+                        {
+                            "input": [{
+                                "type": "image_url",
+                                "url": "test_image_1"
+                            },
+                            {
+                                "type": "image_url",
+                                "url": "test_image_2"
+                            }]
+                        }
+                    ]
+                },
+            ]
         }
-
+        
         assert result == expected_result