Merge pull request #600 from roboflow/fix/fix_platform_tests

Fix integration tests
roboflow · Aug 23, 2024 · eef8e9d · eef8e9d
2 parents bbf64e1 + 2071b19
commit eef8e9d
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/inference/hosted_platform_tests/test_roboflow_models.py b/tests/inference/hosted_platform_tests/test_roboflow_models.py
@@ -593,6 +593,7 @@ def test_infer_from_classification_model_when_valid_response_expected(
     assert set(response.keys()) == {
         "image",
         "predictions",
+        "inference_id",
         "predicted_classes",
         "time",
     }, "Expected all required keys to be provided in response"
@@ -657,6 +658,7 @@ def test_infer_from_classification_model_when_valid_response_expected_with_visua
     assert set(response.keys()) == {
         "visualization",
         "image",
+        "inference_id",
         "predictions",
         "predicted_classes",
         "time",