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 9a48cd4..05c067a 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()) 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