diff --git a/supervision/dataset/formats/pascal_voc.py b/supervision/dataset/formats/pascal_voc.py index 32ba7a74..ba77ddf8 100644 --- a/supervision/dataset/formats/pascal_voc.py +++ b/supervision/dataset/formats/pascal_voc.py @@ -271,7 +271,8 @@ def detections_from_xml_obj( annotation = Detections( xyxy=xyxy.astype(np.float32), mask=np.array(masks).astype(bool) if with_masks else None, - class_id=class_id) + class_id=class_id, + ) return annotation, extended_classes diff --git a/test/dataset/formats/test_pascal_voc.py b/test/dataset/formats/test_pascal_voc.py index fa58f257..23aa65a4 100644 --- a/test/dataset/formats/test_pascal_voc.py +++ b/test/dataset/formats/test_pascal_voc.py @@ -1,4 +1,5 @@ from contextlib import ExitStack as DoesNotRaise +from test.test_utils import mock_detections from typing import List, Optional import defusedxml.ElementTree as ET @@ -6,10 +7,10 @@ import numpy as np import pytest from supervision.dataset.formats.pascal_voc import ( + detections_from_xml_obj, object_to_pascal_voc, - parse_polygon_points, detections_from_xml_obj, + parse_polygon_points, ) -from test.test_utils import mock_detections def are_xml_elements_equal(elem1, elem2): @@ -130,8 +131,7 @@ NO_DETECTIONS = """""" (100, 100), False, mock_detections( - xyxy=np.array([[0, 0, 10, 10], [10, 10, 20, 20]]), - class_id=[0, 0] + xyxy=np.array([[0, 0, 10, 10], [10, 10, 20, 20]]), class_id=[0, 0] ), DoesNotRaise(), ),