From 05eb2f3e5c63eca56f82721782261ca25a37663e Mon Sep 17 00:00:00 2001 From: starkillerOG Date: Thu, 18 Nov 2021 22:52:57 +0100 Subject: [PATCH] Yeelight add state check to set_mode service (#59884) --- homeassistant/components/yeelight/light.py | 1 + 1 file changed, 1 insertion(+) diff --git a/homeassistant/components/yeelight/light.py b/homeassistant/components/yeelight/light.py index e71b75755c0..ca2752a7a7d 100644 --- a/homeassistant/components/yeelight/light.py +++ b/homeassistant/components/yeelight/light.py @@ -842,6 +842,7 @@ class YeelightGenericLight(YeelightEntity, LightEntity): async def async_set_mode(self, mode: str): """Set a power mode.""" await self._bulb.async_set_power_mode(PowerMode[mode.upper()]) + self._async_schedule_state_check(True) @_async_cmd async def async_start_flow(self, transitions, count=0, action=ACTION_RECOVER):