From 3e01943f4f726f636e9e9fcc738d8a8a41f5fae0 Mon Sep 17 00:00:00 2001 From: epenet <6771947+epenet@users.noreply.github.com> Date: Sat, 23 Apr 2022 18:30:07 +0000 Subject: [PATCH] Use ColorMode enum in evil_genius_labs --- homeassistant/components/evil_genius_labs/light.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/homeassistant/components/evil_genius_labs/light.py b/homeassistant/components/evil_genius_labs/light.py index 8233277c09bb9..41fbcfa9b489c 100644 --- a/homeassistant/components/evil_genius_labs/light.py +++ b/homeassistant/components/evil_genius_labs/light.py @@ -6,7 +6,7 @@ from async_timeout import timeout from homeassistant.components import light -from homeassistant.components.light import LightEntity, LightEntityFeature +from homeassistant.components.light import ColorMode, LightEntity, LightEntityFeature from homeassistant.config_entries import ConfigEntry from homeassistant.core import HomeAssistant from homeassistant.helpers.entity_platform import AddEntitiesCallback @@ -33,8 +33,8 @@ class EvilGeniusLight(EvilGeniusEntity, LightEntity): """Evil Genius Labs light.""" _attr_supported_features = LightEntityFeature.EFFECT - _attr_supported_color_modes = {light.COLOR_MODE_RGB} - _attr_color_mode = light.COLOR_MODE_RGB + _attr_supported_color_modes = {ColorMode.RGB} + _attr_color_mode = ColorMode.RGB def __init__(self, coordinator: EvilGeniusUpdateCoordinator) -> None: """Initialize the Evil Genius light."""