Update tests

humanprotocol · Jan 9, 2025 · 1ea84dc · 1ea84dc
1 parent 8974a68
commit 1ea84dc
Showing 1 changed file with 21 additions and 0 deletions.
diff --git a/...ages/examples/cvat/recording-oracle/tests/integration/services/test_validation_service.py b/...ages/examples/cvat/recording-oracle/tests/integration/services/test_validation_service.py
@@ -808,6 +808,13 @@ def test_can_exclude_bad_gt_for_each_label_separately(self, session: Session):
                 frames=[SimpleNamespace(name=f"frame_{i}.jpg") for i in range(frame_count)],
             )
 
+            common_lock_es.enter_context(
+                mock.patch(
+                    "src.handlers.process_intermediate_results.cvat_api.get_task_quality_settings",
+                    return_value=cvat_api.QualitySettings(target_metric="accuracy"),
+                )
+            )
+
             def patched_get_task_labels(task_id: int):
                 return [manifest.annotation.labels[0].nodes[task_id - 1]]
 
@@ -1009,6 +1016,13 @@ def test_can_complete_if_not_enough_gt_left_in_task(
                 frames=[SimpleNamespace(name=f"frame_{i}.jpg") for i in range(frame_count)],
             )
 
+            common_lock_es.enter_context(
+                mock.patch(
+                    "src.handlers.process_intermediate_results.cvat_api.get_task_quality_settings",
+                    return_value=cvat_api.QualitySettings(target_metric="accuracy"),
+                )
+            )
+
             def patched_get_task_labels(task_id: int):
                 return [manifest.annotation.labels[0].nodes[task_id - 1]]
 
@@ -1185,6 +1199,13 @@ def test_can_complete_if_not_enough_gt_left_in_task(
                 frames=[SimpleNamespace(name=f"frame_{i}.jpg") for i in range(frame_count)],
             )
 
+            common_lock_es.enter_context(
+                mock.patch(
+                    "src.handlers.process_intermediate_results.cvat_api.get_task_quality_settings",
+                    return_value=cvat_api.QualitySettings(target_metric="accuracy"),
+                )
+            )
+
             def patched_get_task_labels(task_id: int):
                 return [manifest.annotation.labels[0].nodes[task_id - 1]]