Add Obstacle type hint #16

MissouriMRR · Sep 27, 2022 · f3d0a64 · f3d0a64
1 parent 952b129
commit f3d0a64
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/flight/avoidance/obstacle_avoidance.py b/flight/avoidance/obstacle_avoidance.py
@@ -7,12 +7,13 @@
 from mavsdk import System
 from mavsdk.telemetry import Position
 
-# TODO: Update type to actual type # pylint: disable=fixme
+# TODO: Update types to the correct types # pylint: disable=fixme
+Obstacle = dict[str, float]
 Coordinate = dict[str, float]
 
 
 async def calculate_avoidance_path(
-    drone: System, obstacle: Coordinate, position: Optional[Position]
+    drone: System, obstacle: Obstacle, position: Optional[Position]
 ) -> list[Coordinate]:
     """
     Given a drone and an obstacle, calculates a path avoiding the obstacle