From 4a95ed9b7fbc69a495a810ba55130be22490989b Mon Sep 17 00:00:00 2001 From: Jan Bouwhuis Date: Fri, 30 Jul 2021 07:10:41 +0200 Subject: [PATCH] Fix Xiaomi-miio switch platform setup (#53739) --- homeassistant/components/xiaomi_miio/switch.py | 9 +++------ 1 file changed, 3 insertions(+), 6 deletions(-) diff --git a/homeassistant/components/xiaomi_miio/switch.py b/homeassistant/components/xiaomi_miio/switch.py index c86c98de34c..bdf3085f236 100644 --- a/homeassistant/components/xiaomi_miio/switch.py +++ b/homeassistant/components/xiaomi_miio/switch.py @@ -218,13 +218,10 @@ async def async_setup_platform(hass, config, async_add_entities, discovery_info= async def async_setup_entry(hass, config_entry, async_add_entities): """Set up the switch from a config entry.""" - if ( - config_entry.data[CONF_FLOW_TYPE] == CONF_GATEWAY - or config_entry.data[CONF_MODEL] == "lumi.acpartner.v3" - ): - await async_setup_other_entry(hass, config_entry, async_add_entities) - else: + if config_entry.data[CONF_MODEL] in MODELS_HUMIDIFIER: await async_setup_coordinated_entry(hass, config_entry, async_add_entities) + else: + await async_setup_other_entry(hass, config_entry, async_add_entities) async def async_setup_coordinated_entry(hass, config_entry, async_add_entities):