diff --git a/spatialyze/video_processor/stages/in_view/in_view.py b/spatialyze/video_processor/stages/in_view/in_view.py index 7e19efb..bfdb2b9 100644 --- a/spatialyze/video_processor/stages/in_view/in_view.py +++ b/spatialyze/video_processor/stages/in_view/in_view.py @@ -48,7 +48,9 @@ def __init__( ): super().__init__() self.distance = distance - assert roadtypes is not None or predicate is not None, "At least one of roadtypes or predicate must be specified" + assert ( + roadtypes is not None or predicate is not None + ), "At least one of roadtypes or predicate must be specified" if roadtypes is not None: assert predicate is None, "Can only except either segment_type or predicate" self.roadtypes = roadtypes if isinstance(roadtypes, list) else [roadtypes] @@ -408,12 +410,10 @@ def visit_BoolOpNode(self, node: BoolOpNode): return F.ignore_roadtype() # Boolean Absorption - e_params = [ - e.params[0] - for e in _exprs - if isinstance(e, CallNode) and e.fn == IS_ROADTYPE - ] - assert all(isinstance(e, LiteralNode) and isinstance(e.value, str) for e in e_params), e_params + e_params = [e.params[0] for e in _exprs if isinstance(e, CallNode) and e.fn == IS_ROADTYPE] + assert all( + isinstance(e, LiteralNode) and isinstance(e.value, str) for e in e_params + ), e_params _is_roadtypes = [ v.lower() for v in (e.value for e in e_params if isinstance(e, LiteralNode)) @@ -444,7 +444,9 @@ def is_absorbed(e: "CallNode | BoolOpNode"): if not all_roadtype: return False exprs = [ee.params[0] for ee in e.exprs if isinstance(ee, CallNode)] - assert all(isinstance(e, LiteralNode) and isinstance(e.value, str) for e in exprs), exprs + assert all( + isinstance(e, LiteralNode) and isinstance(e.value, str) for e in exprs + ), exprs return { v.lower() for v in (ee.value for ee in exprs if isinstance(ee, LiteralNode)) diff --git a/spatialyze/video_processor/stages/segment_trajectory/construct_segment_trajectory.py b/spatialyze/video_processor/stages/segment_trajectory/construct_segment_trajectory.py index 4c74e97..a68bfae 100644 --- a/spatialyze/video_processor/stages/segment_trajectory/construct_segment_trajectory.py +++ b/spatialyze/video_processor/stages/segment_trajectory/construct_segment_trajectory.py @@ -327,7 +327,9 @@ def interpolate_segment(args: "Tuple[int, CameraConfig]"): return before + [middle_segment] + after -def complete_segment_trajectory(road_segment_trajectory: "list[ValidSegmentPoint]", payload: "Payload"): +def complete_segment_trajectory( + road_segment_trajectory: "list[ValidSegmentPoint]", payload: "Payload" +): completed_segment_trajectory: "list[ValidSegmentPoint]" = [road_segment_trajectory[0]] for current_segment, next_segment in zip(