From 69df7bc43d76adda624ab5b5557b57b319128309 Mon Sep 17 00:00:00 2001 From: epenet <6771947+epenet@users.noreply.github.com> Date: Mon, 20 Dec 2021 20:59:50 +0100 Subject: [PATCH] Use new enums in ted5000 (#62425) Co-authored-by: epenet --- homeassistant/components/ted5000/sensor.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/homeassistant/components/ted5000/sensor.py b/homeassistant/components/ted5000/sensor.py index a7162ee9c63..c6aedbb5046 100644 --- a/homeassistant/components/ted5000/sensor.py +++ b/homeassistant/components/ted5000/sensor.py @@ -9,8 +9,8 @@ import xmltodict from homeassistant.components.sensor import ( PLATFORM_SCHEMA, - STATE_CLASS_MEASUREMENT, SensorEntity, + SensorStateClass, ) from homeassistant.const import ( CONF_HOST, @@ -62,7 +62,7 @@ def setup_platform(hass, config, add_entities, discovery_info=None): class Ted5000Sensor(SensorEntity): """Implementation of a Ted5000 sensor.""" - _attr_state_class = STATE_CLASS_MEASUREMENT + _attr_state_class = SensorStateClass.MEASUREMENT def __init__(self, gateway, name, mtu, unit): """Initialize the sensor."""