From addf73d007386018eb013076e08f85702aaf4482 Mon Sep 17 00:00:00 2001 From: epenet <6771947+epenet@users.noreply.github.com> Date: Sat, 23 Apr 2022 21:21:44 +0200 Subject: [PATCH] Use ColorMode enum in litejet (#70516) --- homeassistant/components/litejet/light.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/homeassistant/components/litejet/light.py b/homeassistant/components/litejet/light.py index ad638653a94..64d2706565d 100644 --- a/homeassistant/components/litejet/light.py +++ b/homeassistant/components/litejet/light.py @@ -4,8 +4,8 @@ import logging from homeassistant.components.light import ( ATTR_BRIGHTNESS, ATTR_TRANSITION, - COLOR_MODE_BRIGHTNESS, SUPPORT_TRANSITION, + ColorMode, LightEntity, ) from homeassistant.config_entries import ConfigEntry @@ -41,8 +41,8 @@ async def async_setup_entry( class LiteJetLight(LightEntity): """Representation of a single LiteJet light.""" - _attr_color_mode = COLOR_MODE_BRIGHTNESS - _attr_supported_color_modes = {COLOR_MODE_BRIGHTNESS} + _attr_color_mode = ColorMode.BRIGHTNESS + _attr_supported_color_modes = {ColorMode.BRIGHTNESS} _attr_supported_features = SUPPORT_TRANSITION def __init__(self, config_entry, lj, i, name): # pylint: disable=invalid-name