style: [CI] format

apperception-db · Aug 14, 2023 · 34ddf24 · 34ddf24
1 parent 05bb7e6
commit 34ddf24
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 9 deletions.
diff --git 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
@@ -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(