From 63ccc98d0b683adff58c5f70509dbf8f2f8afbae Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Fri, 29 Dec 2023 14:15:59 +0000 Subject: [PATCH] =?UTF-8?q?fix(pre=5Fcommit):=20=F0=9F=8E=A8=20auto=20form?= =?UTF-8?q?at=20pre-commit=20hooks?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- supervision/dataset/formats/pascal_voc.py | 3 ++- test/dataset/formats/test_pascal_voc.py | 8 ++++---- 2 files changed, 6 insertions(+), 5 deletions(-) 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(), ),