From 213e21e8424aafd50242e77bcedc39f0a4b50074 Mon Sep 17 00:00:00 2001 From: Tim Empringham Date: Tue, 28 Feb 2023 12:34:46 -0500 Subject: [PATCH] Add state_class to aurora sensor. --- homeassistant/components/aurora/sensor.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/homeassistant/components/aurora/sensor.py b/homeassistant/components/aurora/sensor.py index 2710badb516..484a45e16f0 100644 --- a/homeassistant/components/aurora/sensor.py +++ b/homeassistant/components/aurora/sensor.py @@ -1,5 +1,5 @@ """Support for Aurora Forecast sensor.""" -from homeassistant.components.sensor import SensorEntity +from homeassistant.components.sensor import SensorEntity, SensorStateClass from homeassistant.config_entries import ConfigEntry from homeassistant.const import PERCENTAGE from homeassistant.core import HomeAssistant @@ -33,3 +33,8 @@ class AuroraSensor(AuroraEntity, SensorEntity): def native_value(self): """Return % chance the aurora is visible.""" return self.coordinator.data + + @property + def state_class(self): + """Return state class for measurement.""" + return SensorStateClass.MEASUREMENT