From 0ddd858b4bd43bab4df6408a99a7bb9c955b09a2 Mon Sep 17 00:00:00 2001 From: Franck Nijhof Date: Wed, 23 Jun 2021 19:43:57 +0200 Subject: [PATCH] Add state class to Aurora ABB Solar PV (#52108) --- homeassistant/components/aurora_abb_powerone/sensor.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/homeassistant/components/aurora_abb_powerone/sensor.py b/homeassistant/components/aurora_abb_powerone/sensor.py index f4640e7c014..cd4a71d1b31 100644 --- a/homeassistant/components/aurora_abb_powerone/sensor.py +++ b/homeassistant/components/aurora_abb_powerone/sensor.py @@ -5,7 +5,11 @@ import logging from aurorapy.client import AuroraError, AuroraSerialClient 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_ADDRESS, CONF_DEVICE, @@ -46,6 +50,8 @@ def setup_platform(hass, config, add_entities, discovery_info=None): class AuroraABBSolarPVMonitorSensor(SensorEntity): """Representation of a Sensor.""" + _attr_state_class = STATE_CLASS_MEASUREMENT + def __init__(self, client, name, typename): """Initialize the sensor.""" self._name = f"{name} {typename}"