diff --git a/mavtypes/waypoint.py b/mavtypes/waypoint.py index 93e2436..0a138d7 100644 --- a/mavtypes/waypoint.py +++ b/mavtypes/waypoint.py @@ -20,3 +20,8 @@ def __hash__(self): round(self.speed, 10), ) ) + + def __iter__(self): + yield self.x + yield self.y + yield self.z diff --git a/protocols/moving_drop_protocol.py b/protocols/moving_drop_protocol.py index 3a9bc02..f0eaa72 100644 --- a/protocols/moving_drop_protocol.py +++ b/protocols/moving_drop_protocol.py @@ -5,7 +5,7 @@ from mavcore.mav_protocol import MAVProtocol from mavcore.messages import SetpointVelocity, LocalPositionNED -from mavcore.types import Waypoint +from mavcore.mavtypes import Waypoint import time import numpy as np from typing import Callable