Update sensor.py

iMicknl · Apr 17, 2024 · ba18462 · ba18462
1 parent 812f51a
commit ba18462
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/custom_components/nest_protect/sensor.py b/custom_components/nest_protect/sensor.py
@@ -11,7 +11,7 @@
     SensorEntity,
     SensorEntityDescription,
 )
-from homeassistant.const import PERCENTAGE, TEMP_CELSIUS
+from homeassistant.const import PERCENTAGE, UnitOfTemperature
 from homeassistant.helpers.entity import EntityCategory
 from homeassistant.helpers.typing import StateType
 
@@ -48,7 +48,7 @@ class NestProtectSensorDescription(SensorEntityDescription):
         key="current_temperature",
         value_fn=lambda state: round(state, 2),
         device_class=SensorDeviceClass.TEMPERATURE,
-        native_unit_of_measurement=TEMP_CELSIUS,
+        native_unit_of_measurement=UnitOfTemperature.CELSIUS,
     ),
     # TODO Add Color Status (gray, green, yellow, red)
     # TODO Smoke Status (OK, Warning, Emergency)