Added typings

CMU-Robotics-Club · Jun 26, 2024 · 4dbd5b0 · 4dbd5b0
1 parent c043e33
commit 4dbd5b0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/rb_ws/src/buggy/scripts/auton/trajectory.py b/rb_ws/src/buggy/scripts/auton/trajectory.py
@@ -349,14 +349,14 @@ def get_closest_index_on_path(
             + start_index
         )
 
-    def pack(self):
+    def pack(self) -> TrajectoryMsg:
         traj = TrajectoryMsg()
         traj.easting = self.positions[:, 0]
         traj.northing = self.positions[:, 1]
         traj.time = time.time()
         return traj
 
-    def unpack(trajMsg : TrajectoryMsg):
+    def unpack(trajMsg : TrajectoryMsg) -> Trajectory:
         pos = np.array([trajMsg.easting, trajMsg.northing]).transpose(1, 0)
         return Trajectory(positions=pos)