From e5e8c71390d3c680aa8dbf32b2a27c7c18e36d87 Mon Sep 17 00:00:00 2001 From: Kinar R <42828719+kinaryml@users.noreply.github.com> Date: Sat, 15 Apr 2023 00:29:33 +0530 Subject: [PATCH] Update object_detector_test.py --- mediapipe/tasks/python/test/vision/object_detector_test.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mediapipe/tasks/python/test/vision/object_detector_test.py b/mediapipe/tasks/python/test/vision/object_detector_test.py index 7a7390051..58b549c12 100644 --- a/mediapipe/tasks/python/test/vision/object_detector_test.py +++ b/mediapipe/tasks/python/test/vision/object_detector_test.py @@ -155,7 +155,7 @@ class ObjectDetectorTest(parameterized.TestCase): ) def _assert_approximately_equals_detection_result(self, result, expected_result): - self.assertLen(result.detections, 4) + self.assertLen(result.detections, len(expected_result.detections)) for i, detection in enumerate(result.detections): bbox = detection.bounding_box category = detection.categories[0]