diff --git a/homeassistant/components/purpleair/sensor.py b/homeassistant/components/purpleair/sensor.py index 2173c67d2d27..e0108d74cb7c 100644 --- a/homeassistant/components/purpleair/sensor.py +++ b/homeassistant/components/purpleair/sensor.py @@ -13,11 +13,11 @@ from homeassistant.components.sensor import ( SensorStateClass, ) from homeassistant.const import ( - CONCENTRATION_MICROGRAMS_PER_CUBIC_METER, - PERCENTAGE, SIGNAL_STRENGTH_DECIBELS_MILLIWATT, EntityCategory, + UnitOfDensity, UnitOfPressure, + UnitOfRatio, UnitOfTemperature, UnitOfTime, UnitOfVolume, @@ -43,7 +43,7 @@ SENSOR_DESCRIPTIONS = [ PurpleAirSensorEntityDescription( key="humidity", device_class=SensorDeviceClass.HUMIDITY, - native_unit_of_measurement=PERCENTAGE, + native_unit_of_measurement=UnitOfRatio.PERCENTAGE, state_class=SensorStateClass.MEASUREMENT, value_fn=lambda sensor: sensor.humidity, ), @@ -74,7 +74,7 @@ SENSOR_DESCRIPTIONS = [ PurpleAirSensorEntityDescription( key="pm1.0_mass_concentration", device_class=SensorDeviceClass.PM1, - native_unit_of_measurement=CONCENTRATION_MICROGRAMS_PER_CUBIC_METER, + native_unit_of_measurement=UnitOfDensity.MICROGRAMS_PER_CUBIC_METER, state_class=SensorStateClass.MEASUREMENT, value_fn=lambda sensor: sensor.pm1_0, ), @@ -89,7 +89,7 @@ SENSOR_DESCRIPTIONS = [ PurpleAirSensorEntityDescription( key="pm10.0_mass_concentration", device_class=SensorDeviceClass.PM10, - native_unit_of_measurement=CONCENTRATION_MICROGRAMS_PER_CUBIC_METER, + native_unit_of_measurement=UnitOfDensity.MICROGRAMS_PER_CUBIC_METER, state_class=SensorStateClass.MEASUREMENT, value_fn=lambda sensor: sensor.pm10_0, ), @@ -104,7 +104,7 @@ SENSOR_DESCRIPTIONS = [ PurpleAirSensorEntityDescription( key="pm2.5_mass_concentration", device_class=SensorDeviceClass.PM25, - native_unit_of_measurement=CONCENTRATION_MICROGRAMS_PER_CUBIC_METER, + native_unit_of_measurement=UnitOfDensity.MICROGRAMS_PER_CUBIC_METER, state_class=SensorStateClass.MEASUREMENT, value_fn=lambda sensor: sensor.pm2_5, ),