From 7ab0dc51777b88d271ddbb1beede48a72e167880 Mon Sep 17 00:00:00 2001 From: epenet <6771947+epenet@users.noreply.github.com> Date: Tue, 26 Apr 2022 14:33:02 +0200 Subject: [PATCH] Use light enums in wiz (#70785) --- homeassistant/components/wiz/light.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/homeassistant/components/wiz/light.py b/homeassistant/components/wiz/light.py index d534b62cf21..084acbe73e7 100644 --- a/homeassistant/components/wiz/light.py +++ b/homeassistant/components/wiz/light.py @@ -13,9 +13,9 @@ from homeassistant.components.light import ( ATTR_EFFECT, ATTR_RGBW_COLOR, ATTR_RGBWW_COLOR, - SUPPORT_EFFECT, ColorMode, LightEntity, + LightEntityFeature, ) from homeassistant.config_entries import ConfigEntry from homeassistant.core import HomeAssistant, callback @@ -91,7 +91,7 @@ class WizBulbEntity(WizToggleEntity, LightEntity): self._attr_min_mireds = color_temperature_kelvin_to_mired(kelvin.max) self._attr_max_mireds = color_temperature_kelvin_to_mired(kelvin.min) if bulb_type.features.effect: - self._attr_supported_features = SUPPORT_EFFECT + self._attr_supported_features = LightEntityFeature.EFFECT self._async_update_attrs() @callback