From 614076cdf1db352744816a397b3081868a8cd15a Mon Sep 17 00:00:00 2001 From: Franck Nijhof Date: Mon, 4 Apr 2022 13:28:19 +0200 Subject: [PATCH] Migrate WLED to use EntityFeature enums (#69219) --- homeassistant/components/wled/light.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) 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__(