diff --git a/homeassistant/components/tado/sensor.py b/homeassistant/components/tado/sensor.py index 044241f2be0..872e2cbb42e 100644 --- a/homeassistant/components/tado/sensor.py +++ b/homeassistant/components/tado/sensor.py @@ -1,7 +1,7 @@ """Support for Tado sensors for each zone.""" import logging -from homeassistant.components.sensor import SensorEntity +from homeassistant.components.sensor import STATE_CLASS_MEASUREMENT, SensorEntity from homeassistant.config_entries import ConfigEntry from homeassistant.const import ( DEVICE_CLASS_HUMIDITY, @@ -153,6 +153,13 @@ class TadoHomeSensor(TadoHomeEntity, SensorEntity): return DEVICE_CLASS_TEMPERATURE return None + @property + def state_class(self): + """Return the state class.""" + if self.home_variable in ["outdoor temperature", "solar percentage"]: + return STATE_CLASS_MEASUREMENT + return None + @callback def _async_update_callback(self): """Update and write state.""" @@ -259,6 +266,13 @@ class TadoZoneSensor(TadoZoneEntity, SensorEntity): return DEVICE_CLASS_TEMPERATURE return None + @property + def state_class(self): + """Return the state class.""" + if self.zone_variable in ["ac", "heating", "humidity", "temperature"]: + return STATE_CLASS_MEASUREMENT + return None + @callback def _async_update_callback(self): """Update and write state."""