From dae8b60ec4d0c3a9efda2d09a50725f14f01770e Mon Sep 17 00:00:00 2001 From: Erwin Douna Date: Thu, 25 Jun 2026 19:08:12 +0200 Subject: [PATCH] Tradfri refactor UnitOfs (#174834) --- homeassistant/components/tradfri/sensor.py | 11 +++-------- 1 file changed, 3 insertions(+), 8 deletions(-) diff --git a/homeassistant/components/tradfri/sensor.py b/homeassistant/components/tradfri/sensor.py index ae385898b868..cbdda2bc4bcc 100644 --- a/homeassistant/components/tradfri/sensor.py +++ b/homeassistant/components/tradfri/sensor.py @@ -13,12 +13,7 @@ from homeassistant.components.sensor import ( SensorEntityDescription, SensorStateClass, ) -from homeassistant.const import ( - CONCENTRATION_MICROGRAMS_PER_CUBIC_METER, - PERCENTAGE, - Platform, - UnitOfTime, -) +from homeassistant.const import Platform, UnitOfDensity, UnitOfRatio, UnitOfTime from homeassistant.core import HomeAssistant, callback from homeassistant.helpers import entity_registry as er from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback @@ -62,7 +57,7 @@ SENSOR_DESCRIPTIONS_BATTERY: tuple[TradfriSensorEntityDescription, ...] = ( key="battery_level", device_class=SensorDeviceClass.BATTERY, state_class=SensorStateClass.MEASUREMENT, - native_unit_of_measurement=PERCENTAGE, + native_unit_of_measurement=UnitOfRatio.PERCENTAGE, value=lambda device: cast(int, device.device_info.battery_level), ), ) @@ -73,7 +68,7 @@ SENSOR_DESCRIPTIONS_FAN: tuple[TradfriSensorEntityDescription, ...] = ( key="aqi", translation_key="aqi", state_class=SensorStateClass.MEASUREMENT, - native_unit_of_measurement=CONCENTRATION_MICROGRAMS_PER_CUBIC_METER, + native_unit_of_measurement=UnitOfDensity.MICROGRAMS_PER_CUBIC_METER, value=_get_air_quality, ), TradfriSensorEntityDescription(