From 666e14b11d2172ca04976476256da635eca64296 Mon Sep 17 00:00:00 2001 From: epenet <6771947+epenet@users.noreply.github.com> Date: Mon, 20 Dec 2021 20:57:05 +0100 Subject: [PATCH] Use new enums in wled (#62431) Co-authored-by: epenet --- homeassistant/components/wled/sensor.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/homeassistant/components/wled/sensor.py b/homeassistant/components/wled/sensor.py index 05d92a91e3d..b37238aef64 100644 --- a/homeassistant/components/wled/sensor.py +++ b/homeassistant/components/wled/sensor.py @@ -8,10 +8,10 @@ from datetime import datetime, timedelta from wled import Device as WLEDDevice from homeassistant.components.sensor import ( - STATE_CLASS_MEASUREMENT, SensorDeviceClass, SensorEntity, SensorEntityDescription, + SensorStateClass, ) from homeassistant.config_entries import ConfigEntry from homeassistant.const import ( @@ -51,7 +51,7 @@ SENSORS: tuple[WLEDSensorEntityDescription, ...] = ( name="Estimated Current", native_unit_of_measurement=ELECTRIC_CURRENT_MILLIAMPERE, device_class=SensorDeviceClass.CURRENT, - state_class=STATE_CLASS_MEASUREMENT, + state_class=SensorStateClass.MEASUREMENT, entity_category=EntityCategory.DIAGNOSTIC, value_fn=lambda device: device.info.leds.power, ), @@ -82,7 +82,7 @@ SENSORS: tuple[WLEDSensorEntityDescription, ...] = ( name="Free Memory", icon="mdi:memory", native_unit_of_measurement=DATA_BYTES, - state_class=STATE_CLASS_MEASUREMENT, + state_class=SensorStateClass.MEASUREMENT, entity_category=EntityCategory.DIAGNOSTIC, entity_registry_enabled_default=False, value_fn=lambda device: device.info.free_heap,