From 01dffc344720013ab7f1495ea8f2cf86a08db53e Mon Sep 17 00:00:00 2001 From: epenet <6771947+epenet@users.noreply.github.com> Date: Tue, 26 Apr 2022 14:31:05 +0200 Subject: [PATCH] Use light enums in mystrom (#70792) --- homeassistant/components/mystrom/light.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/homeassistant/components/mystrom/light.py b/homeassistant/components/mystrom/light.py index bfc527c818f..f6f214301df 100644 --- a/homeassistant/components/mystrom/light.py +++ b/homeassistant/components/mystrom/light.py @@ -12,10 +12,9 @@ from homeassistant.components.light import ( ATTR_EFFECT, ATTR_HS_COLOR, PLATFORM_SCHEMA, - SUPPORT_EFFECT, - SUPPORT_FLASH, ColorMode, LightEntity, + LightEntityFeature, ) from homeassistant.const import CONF_HOST, CONF_MAC, CONF_NAME from homeassistant.core import HomeAssistant @@ -73,7 +72,7 @@ class MyStromLight(LightEntity): _attr_color_mode = ColorMode.HS _attr_supported_color_modes = {ColorMode.HS} - _attr_supported_features = SUPPORT_EFFECT | SUPPORT_FLASH + _attr_supported_features = LightEntityFeature.EFFECT | LightEntityFeature.FLASH def __init__(self, bulb, name, mac): """Initialize the light."""