From 49d268c25018b1a876b76a3cc2c7d0976e4ff421 Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Mon, 9 Oct 2023 11:13:47 +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/tracker/byte_tracker/core.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/supervision/tracker/byte_tracker/core.py b/supervision/tracker/byte_tracker/core.py index 56ef4f45..03589c3d 100644 --- a/supervision/tracker/byte_tracker/core.py +++ b/supervision/tracker/byte_tracker/core.py @@ -1,4 +1,4 @@ -from typing import List, Tuple, Optional +from typing import List, Optional, Tuple import numpy as np @@ -11,7 +11,7 @@ from supervision.tracker.byte_tracker.kalman_filter import KalmanFilter class STrack(BaseTrack): shared_kalman = KalmanFilter() - def __init__(self, tlwh, score, class_ids, mask:Optional[np.array] = None): + def __init__(self, tlwh, score, class_ids, mask: Optional[np.array] = None): # wait activate self._tlwh = np.asarray(tlwh, dtype=np.float32) self.kalman_filter = None @@ -305,7 +305,9 @@ class ByteTrack: """Detections""" detections = [ STrack(STrack.tlbr_to_tlwh(tlbr), s, c, m) - for (tlbr, s, c, m) in zip(dets, scores_keep, class_ids_keep, masks_keep) + for (tlbr, s, c, m) in zip( + dets, scores_keep, class_ids_keep, masks_keep + ) ] else: detections = []