From f61079b236d74f520ddf9b6576f7d972d6ac2efc Mon Sep 17 00:00:00 2001 From: Github Actions Bot Date: Thu, 10 Aug 2023 23:57:00 +0000 Subject: [PATCH] style: [CI] format --- .../tracking_3d/from_tracking_2d_and_detection_3d.py | 7 ++++--- spatialyze/video_processor/utils/insert_trajectory.py | 1 + 2 files changed, 5 insertions(+), 3 deletions(-) diff --git a/spatialyze/video_processor/stages/tracking_3d/from_tracking_2d_and_detection_3d.py b/spatialyze/video_processor/stages/tracking_3d/from_tracking_2d_and_detection_3d.py index b3bfbc7..455b351 100644 --- a/spatialyze/video_processor/stages/tracking_3d/from_tracking_2d_and_detection_3d.py +++ b/spatialyze/video_processor/stages/tracking_3d/from_tracking_2d_and_detection_3d.py @@ -17,7 +17,9 @@ def _run(self, payload: "Payload") -> "tuple[bitarray | None, dict[str, list] | detections = Detection3D.get(payload.metadata) assert detections is not None - for k, detection, tracking, frame in zip(payload.keep, detections, trackings, payload.video): + for k, detection, tracking, frame in zip( + payload.keep, detections, trackings, payload.video + ): dets, _, dids = detection if not k or tracking is None or detection is None: metadata.append(dict()) @@ -33,8 +35,7 @@ def _run(self, payload: "Payload") -> "tuple[bitarray | None, dict[str, list] | detection_map = { did: (det, p, pfc) - for det, p, pfc, did - in zip(dets, points, points_from_camera, dids) + for det, p, pfc, did in zip(dets, points, points_from_camera, dids) } trackings3d: "dict[int, Tracking3DResult]" = {} for object_id, t in tracking.items(): diff --git a/spatialyze/video_processor/utils/insert_trajectory.py b/spatialyze/video_processor/utils/insert_trajectory.py index d40f1ca..c45285b 100644 --- a/spatialyze/video_processor/utils/insert_trajectory.py +++ b/spatialyze/video_processor/utils/insert_trajectory.py @@ -1,4 +1,5 @@ import datetime + import numpy as np import numpy.typing as npt