fix inference request test

ai-cfia · Jul 29, 2024 · 997713c · 997713c
1 parent 72678a0
commit 997713c
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tests/test_inference_request.py b/tests/test_inference_request.py
@@ -70,7 +70,9 @@ def test_inference_request_successful(self, mock_container):
             "box_id",
             "inference_id",
             "object_type_id",
-            "top_id"
+            "top_id",
+            "models",
+            "pipeline_id"
         }
 
         # Test the answers from inference_request