Add sensor state class for SleepIQ sensors (#76372)

Co-authored-by: Inca <inca@popre.net>
This commit is contained in:
Keilin Bickar 2022-08-10 17:59:50 -04:00 committed by GitHub
parent 2bab6447a9
commit d81298a2d6
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -3,7 +3,7 @@ from __future__ import annotations
from asyncsleepiq import SleepIQBed, SleepIQSleeper
from homeassistant.components.sensor import SensorEntity
from homeassistant.components.sensor import SensorEntity, SensorStateClass
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.entity_platform import AddEntitiesCallback
@ -45,6 +45,7 @@ class SleepIQSensorEntity(SleepIQSleeperEntity, SensorEntity):
) -> None:
"""Initialize the sensor."""
self.sensor_type = sensor_type
self._attr_state_class = SensorStateClass.MEASUREMENT
super().__init__(coordinator, bed, sleeper, sensor_type)
@callback