Fix return to base logic for neato (#16776)

This commit is contained in:
Daniel Shokouhi 2018-09-22 02:34:46 -07:00 committed by Paulus Schoutsen
parent 5613816476
commit eec6722cf4

View file

@ -188,6 +188,8 @@ class NeatoConnectedVacuum(StateVacuumDevice):
def return_to_base(self, **kwargs):
"""Set the vacuum cleaner to return to the dock."""
if self._clean_state == STATE_CLEANING:
self.robot.pause_cleaning()
self._clean_state = STATE_RETURNING
self.robot.send_to_base()