From 7f7c0febd80847d23ad41851b88b0ee6cbc5ed02 Mon Sep 17 00:00:00 2001 From: Franck Nijhof Date: Wed, 23 Jun 2021 19:44:41 +0200 Subject: [PATCH] Add state class to DTE Energy Bridge (#52110) --- homeassistant/components/dte_energy_bridge/sensor.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/homeassistant/components/dte_energy_bridge/sensor.py b/homeassistant/components/dte_energy_bridge/sensor.py index 27475990de0..4e095955818 100644 --- a/homeassistant/components/dte_energy_bridge/sensor.py +++ b/homeassistant/components/dte_energy_bridge/sensor.py @@ -4,7 +4,11 @@ import logging import requests import voluptuous as vol -from homeassistant.components.sensor import PLATFORM_SCHEMA, SensorEntity +from homeassistant.components.sensor import ( + PLATFORM_SCHEMA, + STATE_CLASS_MEASUREMENT, + SensorEntity, +) from homeassistant.const import CONF_NAME, HTTP_OK import homeassistant.helpers.config_validation as cv @@ -41,6 +45,8 @@ def setup_platform(hass, config, add_entities, discovery_info=None): class DteEnergyBridgeSensor(SensorEntity): """Implementation of the DTE Energy Bridge sensors.""" + _attr_state_class = STATE_CLASS_MEASUREMENT + def __init__(self, ip_address, name, version): """Initialize the sensor.""" self._version = version