From 7711ac901ce46ce2861a240df1d3297312874e8b Mon Sep 17 00:00:00 2001 From: Marc Mueller <30130371+cdce8p@users.noreply.github.com> Date: Tue, 20 Jul 2021 05:22:13 +0200 Subject: [PATCH] Fix issues after pylint update (#53205) --- homeassistant/components/xiaomi_miio/fan.py | 4 +--- homeassistant/components/xiaomi_miio/switch.py | 2 +- 2 files changed, 2 insertions(+), 4 deletions(-) diff --git a/homeassistant/components/xiaomi_miio/fan.py b/homeassistant/components/xiaomi_miio/fan.py index 49ae1cf7b7c..c1dde7ec38d 100644 --- a/homeassistant/components/xiaomi_miio/fan.py +++ b/homeassistant/components/xiaomi_miio/fan.py @@ -635,9 +635,7 @@ async def async_setup_entry(hass, config_entry, async_add_entities): await asyncio.wait(update_tasks) for air_purifier_service, method in SERVICE_TO_METHOD.items(): - schema = method[air_purifier_service].get( - "schema", AIRPURIFIER_SERVICE_SCHEMA - ) + schema = method.get("schema", AIRPURIFIER_SERVICE_SCHEMA) hass.services.async_register( DOMAIN, air_purifier_service, async_service_handler, schema=schema ) diff --git a/homeassistant/components/xiaomi_miio/switch.py b/homeassistant/components/xiaomi_miio/switch.py index 52682a05f08..5802ae2a00d 100644 --- a/homeassistant/components/xiaomi_miio/switch.py +++ b/homeassistant/components/xiaomi_miio/switch.py @@ -251,7 +251,7 @@ async def async_setup_entry(hass, config_entry, async_add_entities): await asyncio.wait(update_tasks) for plug_service, method in SERVICE_TO_METHOD.items(): - schema = method[plug_service].get("schema", SERVICE_SCHEMA) + schema = method.get("schema", SERVICE_SCHEMA) hass.services.async_register( DOMAIN, plug_service, async_service_handler, schema=schema )