Fix tests for 2025.1 (#207)

MatthewFlamm · Dec 25, 2024 · 4f5b999 · 4f5b999
1 parent 939a754
commit 4f5b999
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/custom_components/simple_integration/sensor.py b/custom_components/simple_integration/sensor.py
@@ -1,5 +1,5 @@
 """Platform for sensor integration."""
-from homeassistant.const import TEMP_CELSIUS
+from homeassistant.const import UnitOfTemperature
 from homeassistant.helpers.entity import Entity
 
 
@@ -33,4 +33,4 @@ def state(self):
     @property
     def unit_of_measurement(self):
         """Return the unit of measurement."""
-        return TEMP_CELSIUS
+        return UnitOfTemperature.CELSIUS