fix type

spatialyze/video_processor/utils/insert_trajectory.py

@@ -6,16 +6,16 @@
 from psycopg2.sql import SQL, Literal
 
 from ..types import Float3
-from ..utils.prepare_trajectory import Trajectory
 from .infer_heading import infer_heading
 
 if TYPE_CHECKING:
     from ...database import Database
+    from .types import Trajectory
 
 
 def insert_trajectory(
     database: "Database",
-    trajectory: Trajectory,
+    trajectory: "Trajectory",
 ):
     (
         item_id,

spatialyze/video_processor/utils/prepare_trajectory.py

@@ -1,18 +1,9 @@
 import datetime
-from typing import NamedTuple
 
 from ..camera_config import CameraConfig
 from ..stream.strongsort import TrackingResult
 from ..types import Float3
-
-
-class Trajectory(NamedTuple):
-    obj_id: int | str
-    ids: list[int]
-    camera_id: str
-    object_type: str
-    pairs: list[Float3]
-    itemHeadings: list[float | None]
+from .types import Trajectory
 
 
 def prepare_trajectory(

spatialyze/video_processor/utils/types.py

@@ -0,0 +1,12 @@
+from typing import NamedTuple
+
+from ..types import Float3
+
+
+class Trajectory(NamedTuple):
+    obj_id: int | str
+    ids: list[int]
+    camera_id: str
+    object_type: str
+    pairs: list[Float3]
+    itemHeadings: list[float | None]