From d668c9549570d5ea56c7c28a887ad5938d9c95d5 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Pawe=C5=82=20P=C4=99czek?= Date: Thu, 22 Aug 2024 19:58:07 +0200 Subject: [PATCH] Fix integration tests --- tests/inference/hosted_platform_tests/test_roboflow_models.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tests/inference/hosted_platform_tests/test_roboflow_models.py b/tests/inference/hosted_platform_tests/test_roboflow_models.py index 0069d6f215..495e5a171a 100644 --- 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",