From 33b7861c7080e216c409066969718e50337b1dad Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Tue, 30 Jan 2024 17:36:03 +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/detection/core.py | 8 +++----- 1 file changed, 3 insertions(+), 5 deletions(-) diff --git a/supervision/detection/core.py b/supervision/detection/core.py index 62ed8869..dc730989 100644 --- a/supervision/detection/core.py +++ b/supervision/detection/core.py @@ -5,8 +5,8 @@ from dataclasses import dataclass, field from typing import Any, Dict, Iterator, List, Optional, Tuple, Union import numpy as np -from supervision.config import CLASS_NAME_DATA_FIELD, ORIENTED_BOX_COORDINATES +from supervision.config import CLASS_NAME_DATA_FIELD, ORIENTED_BOX_COORDINATES from supervision.detection.utils import ( calculate_masks_centroids, extract_ultralytics_masks, @@ -192,7 +192,7 @@ class Detections: else None, data={ ORIENTED_BOX_COORDINATES: oriented_box_coordinates, - CLASS_NAME_DATA_FIELD: class_names + CLASS_NAME_DATA_FIELD: class_names, }, ) @@ -206,9 +206,7 @@ class Detections: tracker_id=ultralytics_results.boxes.id.int().cpu().numpy() if ultralytics_results.boxes.id is not None else None, - data={ - CLASS_NAME_DATA_FIELD: class_names - } + data={CLASS_NAME_DATA_FIELD: class_names}, ) @classmethod