From 927b74b4a2de611a18856ef2fe6d6f90d24dce3b Mon Sep 17 00:00:00 2001 From: Franck Nijhof Date: Wed, 23 Jun 2021 19:44:25 +0200 Subject: [PATCH] Add state class to The Energy Detective TED5000 (#52109) --- homeassistant/components/ted5000/sensor.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/homeassistant/components/ted5000/sensor.py b/homeassistant/components/ted5000/sensor.py index 62cdd5066ad..5c439651ed5 100644 --- a/homeassistant/components/ted5000/sensor.py +++ b/homeassistant/components/ted5000/sensor.py @@ -7,7 +7,11 @@ import requests import voluptuous as vol import xmltodict -from homeassistant.components.sensor import PLATFORM_SCHEMA, SensorEntity +from homeassistant.components.sensor import ( + PLATFORM_SCHEMA, + STATE_CLASS_MEASUREMENT, + SensorEntity, +) from homeassistant.const import CONF_HOST, CONF_NAME, CONF_PORT, POWER_WATT, VOLT from homeassistant.helpers import config_validation as cv from homeassistant.util import Throttle @@ -52,6 +56,8 @@ def setup_platform(hass, config, add_entities, discovery_info=None): class Ted5000Sensor(SensorEntity): """Implementation of a Ted5000 sensor.""" + _attr_state_class = STATE_CLASS_MEASUREMENT + def __init__(self, gateway, name, mtu, unit): """Initialize the sensor.""" units = {POWER_WATT: "power", VOLT: "voltage"}