From b593c1ed151a9016caf8910e84661b1c989c8199 Mon Sep 17 00:00:00 2001
From: "pre-commit-ci[bot]"
 <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Date: Mon, 16 Dec 2024 17:56:49 +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/yolo.py | 15 +++++++--------
 1 file changed, 7 insertions(+), 8 deletions(-)

diff --git a/supervision/dataset/formats/yolo.py b/supervision/dataset/formats/yolo.py
index ba050c7ec..185154348 100644
--- a/supervision/dataset/formats/yolo.py
+++ b/supervision/dataset/formats/yolo.py
@@ -8,7 +8,7 @@
 from supervision.config import ORIENTED_BOX_COORDINATES
 from supervision.dataset.utils import approximate_mask_with_polygons
 from supervision.detection.core import Detections
-from supervision.detection.utils import polygon_to_mask, polygon_to_xyxy
+from supervision.detection.utils import polygon_to_xyxy
 from supervision.utils.file import (
     list_files_with_extensions,
     read_txt_file,
@@ -44,16 +44,17 @@ def _parse_polygon(values: List[str]) -> np.ndarray:
     return np.array(values, dtype=np.float32).reshape(-1, 2)
 
 
-def _polygons_to_masks(polygon: list[np.ndarray], resolution_wh: Tuple[int, int]) -> np.ndarray:
+def _polygons_to_masks(
+    polygon: list[np.ndarray], resolution_wh: Tuple[int, int]
+) -> np.ndarray:
     polygon_int = np.round(polygon).astype(np.int32)
     mask = np.zeros((resolution_wh[1], resolution_wh[0]), dtype=np.uint8)
-    
+
     cv2.fillPoly(mask, [polygon_int], 1)
 
     return mask.astype(bool)
 
 
-
 def _with_mask(lines: List[str]) -> bool:
     return any([len(line.split()) > 5 for line in lines])
 
@@ -112,9 +113,7 @@ def yolo_annotations_to_detections(
     if not with_masks:
         return Detections(class_id=class_id, xyxy=xyxy, data=data)
 
-    polygons = [
-        (polygon * np.array(resolution_wh)) for polygon in relative_polygon
-    ]
+    polygons = [(polygon * np.array(resolution_wh)) for polygon in relative_polygon]
     mask = _polygons_to_masks(polygon=polygons, resolution_wh=resolution_wh)
     return Detections(class_id=class_id, xyxy=xyxy, data=data, mask=mask)
 
@@ -143,7 +142,7 @@ def load_yolo_annotations(
             where pairs of [x, y] are box corners.
 
     Returns:
-    
+
         Tuple[List[str], List[str], Dict[str, Detections]]:
             A tuple containing a list of class names, a dictionary with
             image names as keys and images as values, and a dictionary