Add state_class to aurora sensor.

This commit is contained in:
Tim Empringham
2023-02-28 12:34:46 -05:00
parent a3546ad88d
commit 213e21e842

View File

@@ -1,5 +1,5 @@
"""Support for Aurora Forecast sensor.""" """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.config_entries import ConfigEntry
from homeassistant.const import PERCENTAGE from homeassistant.const import PERCENTAGE
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant
@@ -33,3 +33,8 @@ class AuroraSensor(AuroraEntity, SensorEntity):
def native_value(self): def native_value(self):
"""Return % chance the aurora is visible.""" """Return % chance the aurora is visible."""
return self.coordinator.data return self.coordinator.data
@property
def state_class(self):
"""Return state class for measurement."""
return SensorStateClass.MEASUREMENT