From 2d1c6ecad8eec3db2164f0f240e55288f5561bc3 Mon Sep 17 00:00:00 2001 From: Michael Graf Date: Tue, 28 Dec 2021 16:43:33 +0100 Subject: [PATCH] Fix vacuum won't start when docked #4 --- vacuum.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/vacuum.py b/vacuum.py index 003e2d9..9fda1f7 100644 --- a/vacuum.py +++ b/vacuum.py @@ -156,7 +156,7 @@ class VorwerkConnectedVacuum(CoordinatorEntity, StateVacuumEntity): if not self._state: return try: - if self._state.state == STATE_IDLE: + if self._state.state == STATE_IDLE or self._state.state == STATE_DOCKED: self.robot.start_cleaning() elif self._state.state == STATE_PAUSED: self.robot.resume_cleaning() @@ -177,7 +177,7 @@ class VorwerkConnectedVacuum(CoordinatorEntity, StateVacuumEntity): def return_to_base(self, **kwargs: Any) -> None: """Set the vacuum cleaner to return to the dock.""" try: - if self._state.state == STATE_CLEANING or self._state.state == STATE_DOCKED: + if self._state.state == STATE_CLEANING: self.robot.pause_cleaning() self.robot.send_to_base() except NeatoRobotException as ex: