Add type asserts to fix the rest of mypy errors

LeoRover · Jun 26, 2024 · f5815d8 · f5815d8
1 parent 6880965
commit f5815d8
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/leo_docking/leo_docking/states/reach_docking_area.py b/leo_docking/leo_docking/states/reach_docking_area.py
@@ -161,13 +161,17 @@ def _get_speed(
         self, route_left: float, route_done: float, direction: float
     ) -> float:
         if self.angle:
+            assert isinstance(self.params, RotateToDockAreaParams) or isinstance(
+                self.params, RotateToBoardParams
+            )
             return direction * translate(
                 route_left - route_done,
                 self.params.angle_min,
                 self.params.angle_max,
                 self.params.speed_min,
                 self.params.speed_max,
             )
+        assert isinstance(self.params, RideToDockArea)
         return direction * translate(
             route_left - route_done,
             self.params.dist_min,