diff --git a/homeassistant/components/nanoleaf/light.py b/homeassistant/components/nanoleaf/light.py index 7cbb48fbf61..d2380f17d18 100644 --- a/homeassistant/components/nanoleaf/light.py +++ b/homeassistant/components/nanoleaf/light.py @@ -12,10 +12,9 @@ from homeassistant.components.light import ( ATTR_EFFECT, ATTR_HS_COLOR, ATTR_TRANSITION, - COLOR_MODE_COLOR_TEMP, - COLOR_MODE_HS, SUPPORT_EFFECT, SUPPORT_TRANSITION, + ColorMode, LightEntity, ) from homeassistant.config_entries import ConfigEntry @@ -46,7 +45,7 @@ async def async_setup_entry( class NanoleafLight(NanoleafEntity, LightEntity): """Representation of a Nanoleaf Light.""" - _attr_supported_color_modes = {COLOR_MODE_COLOR_TEMP, COLOR_MODE_HS} + _attr_supported_color_modes = {ColorMode.COLOR_TEMP, ColorMode.HS} _attr_supported_features = SUPPORT_EFFECT | SUPPORT_TRANSITION def __init__(self, nanoleaf: Nanoleaf, coordinator: DataUpdateCoordinator) -> None: @@ -99,14 +98,14 @@ class NanoleafLight(NanoleafEntity, LightEntity): return self._nanoleaf.hue, self._nanoleaf.saturation @property - def color_mode(self) -> str | None: + def color_mode(self) -> ColorMode | None: """Return the color mode of the light.""" # According to API docs, color mode is "ct", "effect" or "hs" # https://forum.nanoleaf.me/docs/openapi#_4qgqrz96f44d if self._nanoleaf.color_mode == "ct": - return COLOR_MODE_COLOR_TEMP + return ColorMode.COLOR_TEMP # Home Assistant does not have an "effect" color mode, just report hs - return COLOR_MODE_HS + return ColorMode.HS async def async_turn_on(self, **kwargs: Any) -> None: """Instruct the light to turn on."""