diff --git a/homeassistant/components/wled/light.py b/homeassistant/components/wled/light.py index b30e20810d9..93aa95e1a80 100644 --- a/homeassistant/components/wled/light.py +++ b/homeassistant/components/wled/light.py @@ -13,9 +13,8 @@ from homeassistant.components.light import ( COLOR_MODE_BRIGHTNESS, COLOR_MODE_RGB, COLOR_MODE_RGBW, - SUPPORT_EFFECT, - SUPPORT_TRANSITION, LightEntity, + LightEntityFeature, ) from homeassistant.config_entries import ConfigEntry from homeassistant.core import HomeAssistant, callback @@ -55,7 +54,7 @@ class WLEDMasterLight(WLEDEntity, LightEntity): _attr_color_mode = COLOR_MODE_BRIGHTNESS _attr_icon = "mdi:led-strip-variant" - _attr_supported_features = SUPPORT_TRANSITION + _attr_supported_features = LightEntityFeature.TRANSITION def __init__(self, coordinator: WLEDDataUpdateCoordinator) -> None: """Initialize WLED master light.""" @@ -105,7 +104,7 @@ class WLEDMasterLight(WLEDEntity, LightEntity): class WLEDSegmentLight(WLEDEntity, LightEntity): """Defines a WLED light based on a segment.""" - _attr_supported_features = SUPPORT_EFFECT | SUPPORT_TRANSITION + _attr_supported_features = LightEntityFeature.EFFECT | LightEntityFeature.TRANSITION _attr_icon = "mdi:led-strip-variant" def __init__(