evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Use enums in statistics tests (#62191)

Browse source
This commit is contained in:
Robert Hillis 2021-12-18 08:54:26 -05:00 • committed by GitHub
parent a39f0643e8
commit af631b90e5
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 6 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

4
homeassistant/components/statistics/sensor.py
View file

@ -10,8 +10,8 @@ from homeassistant.components.recorder.models import States
from homeassistant.components.recorder.util import execute, session_scope
from homeassistant.components.sensor import (
PLATFORM_SCHEMA,
STATE_CLASS_MEASUREMENT,
SensorEntity,
SensorStateClass,
)
from homeassistant.const import (
ATTR_UNIT_OF_MEASUREMENT,
@ -350,7 +350,7 @@ class StatisticsSensor(SensorEntity):
"""Return the state class of this entity."""
if self._state_characteristic in STATS_NOT_A_NUMBER:
return None
return STATE_CLASS_MEASUREMENT
return SensorStateClass.MEASUREMENT
@property
def native_value(self):