diff --git a/homeassistant/components/mysensors/light.py b/homeassistant/components/mysensors/light.py index 870cde863c4..e2c676220fa 100644 --- a/homeassistant/components/mysensors/light.py +++ b/homeassistant/components/mysensors/light.py @@ -8,9 +8,7 @@ from homeassistant.components.light import ( ATTR_BRIGHTNESS, ATTR_RGB_COLOR, ATTR_RGBW_COLOR, - COLOR_MODE_BRIGHTNESS, - COLOR_MODE_RGB, - COLOR_MODE_RGBW, + ColorMode, LightEntity, ) from homeassistant.config_entries import ConfigEntry @@ -136,8 +134,8 @@ class MySensorsLight(mysensors.device.MySensorsEntity, LightEntity): class MySensorsLightDimmer(MySensorsLight): """Dimmer child class to MySensorsLight.""" - _attr_supported_color_modes = {COLOR_MODE_BRIGHTNESS} - _attr_color_mode = COLOR_MODE_BRIGHTNESS + _attr_supported_color_modes = {ColorMode.BRIGHTNESS} + _attr_color_mode = ColorMode.BRIGHTNESS async def async_turn_on(self, **kwargs: Any) -> None: """Turn the device on.""" @@ -156,8 +154,8 @@ class MySensorsLightDimmer(MySensorsLight): class MySensorsLightRGB(MySensorsLight): """RGB child class to MySensorsLight.""" - _attr_supported_color_modes = {COLOR_MODE_RGB} - _attr_color_mode = COLOR_MODE_RGB + _attr_supported_color_modes = {ColorMode.RGB} + _attr_color_mode = ColorMode.RGB async def async_turn_on(self, **kwargs: Any) -> None: """Turn the device on.""" @@ -202,8 +200,8 @@ class MySensorsLightRGB(MySensorsLight): class MySensorsLightRGBW(MySensorsLightRGB): """RGBW child class to MySensorsLightRGB.""" - _attr_supported_color_modes = {COLOR_MODE_RGBW} - _attr_color_mode = COLOR_MODE_RGBW + _attr_supported_color_modes = {ColorMode.RGBW} + _attr_color_mode = ColorMode.RGBW async def async_turn_on(self, **kwargs: Any) -> None: """Turn the device on."""