diff --git a/homeassistant/components/airvisual_pro/sensor.py b/homeassistant/components/airvisual_pro/sensor.py index 00c87b02377..2708cc5857d 100644 --- a/homeassistant/components/airvisual_pro/sensor.py +++ b/homeassistant/components/airvisual_pro/sensor.py @@ -26,22 +26,15 @@ from . import AirVisualProData, AirVisualProEntity from .const import DOMAIN -@dataclass(frozen=True) -class AirVisualProMeasurementKeyMixin: - """Define an entity description mixin to include a measurement key.""" +@dataclass(frozen=True, kw_only=True) +class AirVisualProMeasurementDescription(SensorEntityDescription): + """Describe an AirVisual Pro sensor.""" value_fn: Callable[ [dict[str, Any], dict[str, Any], dict[str, Any], dict[str, Any]], float | int ] -@dataclass(frozen=True) -class AirVisualProMeasurementDescription( - SensorEntityDescription, AirVisualProMeasurementKeyMixin -): - """Describe an AirVisual Pro sensor.""" - - SENSOR_DESCRIPTIONS = ( AirVisualProMeasurementDescription( key="air_quality_index",