From 33c1501f35a5f02c8a8a21bf3e1baae70c51894c Mon Sep 17 00:00:00 2001 From: epenet <6771947+epenet@users.noreply.github.com> Date: Sat, 23 Apr 2022 21:21:05 +0200 Subject: [PATCH] Use ColorMode enum in futurenow (#70498) --- homeassistant/components/futurenow/light.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/homeassistant/components/futurenow/light.py b/homeassistant/components/futurenow/light.py index 8575017f7e3..e27436966fc 100644 --- a/homeassistant/components/futurenow/light.py +++ b/homeassistant/components/futurenow/light.py @@ -6,9 +6,8 @@ import voluptuous as vol from homeassistant.components.light import ( ATTR_BRIGHTNESS, - COLOR_MODE_BRIGHTNESS, - COLOR_MODE_ONOFF, PLATFORM_SCHEMA, + ColorMode, LightEntity, ) from homeassistant.const import CONF_DEVICES, CONF_HOST, CONF_NAME, CONF_PORT @@ -110,8 +109,8 @@ class FutureNowLight(LightEntity): def color_mode(self) -> str: """Return the color mode of the light.""" if self._dimmable: - return COLOR_MODE_BRIGHTNESS - return COLOR_MODE_ONOFF + return ColorMode.BRIGHTNESS + return ColorMode.ONOFF @property def supported_color_modes(self) -> set[str] | None: