diff --git a/spatialyze/video_processor/utils/insert_trajectory.py b/spatialyze/video_processor/utils/insert_trajectory.py index d57a1c2..53d3bb6 100644 --- a/spatialyze/video_processor/utils/insert_trajectory.py +++ b/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, diff --git a/spatialyze/video_processor/utils/prepare_trajectory.py b/spatialyze/video_processor/utils/prepare_trajectory.py index c92b1d2..003e7e0 100644 --- a/spatialyze/video_processor/utils/prepare_trajectory.py +++ b/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( diff --git a/spatialyze/video_processor/utils/types.py b/spatialyze/video_processor/utils/types.py new file mode 100644 index 0000000..09955f4 --- /dev/null +++ b/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]