fix typing

apperception-db · Aug 10, 2023 · bdcdcb7 · bdcdcb7
1 parent d588eb1
commit bdcdcb7
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 4 deletions.
diff --git a/spatialyze/video_processor/utils/format_trajectory.py b/spatialyze/video_processor/utils/format_trajectory.py
@@ -11,7 +11,7 @@
 
 def format_trajectory(
     video_name: "str",
-    obj_id: "str",
+    obj_id: "int",
     track: "list[tuple[Tracking3DResult, CameraConfig, SegmentPoint | None]]",
     base=None,
 ):

diff --git a/spatialyze/video_processor/utils/insert_trajectory.py b/spatialyze/video_processor/utils/insert_trajectory.py
@@ -1,3 +1,7 @@
+import datetime
+import numpy as np
+import numpy.typing as np
+
 from ...database import Database
 from ...utils import join
 from ..types import Float3
@@ -9,9 +13,9 @@ def insert_trajectory(
     item_id: str,
     camera_id: str,
     object_type: str,
-    postgres_timestamps: "list[str]",
-    pairs: "list[Float3]",
-    itemHeading_list: "list[int]",
+    postgres_timestamps: "list[datetime.datetime]",
+    pairs: "list[npt.NDArray[np.floating]]",
+    itemHeading_list: "list[float | None]",
     translation_list: "list[Float3]",
     # road_types: "list[str]",
     # roadpolygon_list: "list[list[tuple[float, float]]]"