fix tests

octo-technology · Jun 9, 2023 · 7565aee · 7565aee
1 parent 76a4f37
commit 7565aee
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/...tests/integration_tests/infrastructure/model_forward/test_tf_serving_detection_wrapper.py b/...tests/integration_tests/infrastructure/model_forward/test_tf_serving_detection_wrapper.py
@@ -40,7 +40,7 @@ async def test_perform_inference_should_detected_a_cat(self, test_tensorflow_ser
         )
 
         expected_model_output = {
-            'object_1': {'label': 'cat', 'location': [370, 2, 738, 340], 'score': 0.653498411}
+            'object_1': {'label': 'cat', 'location': [370, 2, 738, 340], 'score': 0.652510464}
         }
 
         # When

diff --git a/...s/integration_tests/infrastructure/model_forward/test_tflite_serving_detection_wrapper.py b/...s/integration_tests/infrastructure/model_forward/test_tflite_serving_detection_wrapper.py
@@ -42,7 +42,7 @@ async def test_perform_inference_should_detect_a_bear(
         )
 
         expected_model_output = {
-            'object_1': {'label': 'bear', 'location': [383, 4, 723, 339], 'score': 0.83984375}
+            'object_1': {'label': 'bear', 'location': [376, 4, 722, 339], 'score': 0.87890625}
         }
 
         # When
-Original file line number
+Diff line change
@@ Expand Up @@
             )
             expected_model_output = {
-                'object_1': {'label': 'cat', 'location': [370, 2, 738, 340], 'score': 0.653498411}
+                'object_1': {'label': 'cat', 'location': [370, 2, 738, 340], 'score': 0.652510464}
             }
             # When
@@ Expand Down @@