From 0257f15d6e6b136a0297c1c021fdb5bb2a5b46e4 Mon Sep 17 00:00:00 2001 From: Github Actions Bot Date: Fri, 11 Aug 2023 00:56:19 +0000 Subject: [PATCH] style: [CI] format --- .../stages/tracking_3d/from_tracking_2d_and_detection_3d.py | 5 +++-- .../stages/tracking_3d/from_tracking_2d_and_road.py | 4 +++- 2 files changed, 6 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 05c067a..563bbb0 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 @@ -35,8 +35,9 @@ 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.tolist(), points_from_camera.tolist(), dids) + for det, p, pfc, did in zip( + dets, points.tolist(), points_from_camera.tolist(), dids + ) } trackings3d: "dict[int, Tracking3DResult]" = {} for object_id, t in tracking.items(): diff --git a/spatialyze/video_processor/stages/tracking_3d/from_tracking_2d_and_road.py b/spatialyze/video_processor/stages/tracking_3d/from_tracking_2d_and_road.py index dcf8659..9afc79e 100644 --- a/spatialyze/video_processor/stages/tracking_3d/from_tracking_2d_and_road.py +++ b/spatialyze/video_processor/stages/tracking_3d/from_tracking_2d_and_road.py @@ -63,7 +63,9 @@ def _run(self, payload: "Payload"): points = rotated_directions * ts + translation[:, np.newaxis] points_from_camera = rotate(points - translation[:, np.newaxis], rotation.inverse) - for t, oid, point, point_from_camera in zip(_ts, oids, points.T.tolist(), points_from_camera.T.tolist()): + for t, oid, point, point_from_camera in zip( + _ts, oids, points.T.tolist(), points_from_camera.T.tolist() + ): assert point_from_camera.shape == (3,) assert isinstance(oid, int) or oid.is_integer() oid = int(oid)