diff --git a/homeassistant/components/iqvia/sensor.py b/homeassistant/components/iqvia/sensor.py index d4c01c6fb67..46da1aea0aa 100644 --- a/homeassistant/components/iqvia/sensor.py +++ b/homeassistant/components/iqvia/sensor.py @@ -7,9 +7,9 @@ from typing import NamedTuple import numpy as np from homeassistant.components.sensor import ( - STATE_CLASS_MEASUREMENT, SensorEntity, SensorEntityDescription, + SensorStateClass, ) from homeassistant.config_entries import ConfigEntry from homeassistant.const import ATTR_STATE @@ -99,7 +99,7 @@ INDEX_SENSOR_DESCRIPTIONS = ( key=TYPE_ALLERGY_TODAY, name="Allergy Index: Today", icon="mdi:flower", - state_class=STATE_CLASS_MEASUREMENT, + state_class=SensorStateClass.MEASUREMENT, ), SensorEntityDescription( key=TYPE_ALLERGY_TOMORROW, @@ -110,7 +110,7 @@ INDEX_SENSOR_DESCRIPTIONS = ( key=TYPE_ASTHMA_TODAY, name="Asthma Index: Today", icon="mdi:flower", - state_class=STATE_CLASS_MEASUREMENT, + state_class=SensorStateClass.MEASUREMENT, ), SensorEntityDescription( key=TYPE_ASTHMA_TOMORROW, @@ -121,7 +121,7 @@ INDEX_SENSOR_DESCRIPTIONS = ( key=TYPE_DISEASE_TODAY, name="Cold & Flu Index: Today", icon="mdi:pill", - state_class=STATE_CLASS_MEASUREMENT, + state_class=SensorStateClass.MEASUREMENT, ), )