style: [CI] format

apperception-db · Aug 11, 2023 · 0257f15 · 0257f15
1 parent a04ab08
commit 0257f15
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 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
@@ -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
@@ -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)