Use ColorMode enum in evil_genius_labs (#70493)

This commit is contained in:
epenet 2022-04-23 21:20:09 +02:00 committed by GitHub
parent 80db3428b8
commit d75862762b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -6,7 +6,7 @@ from typing import Any, cast
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."""