Update sensor.py (#28)

Fixes to update deprecated Constants to be removed in HA Core 2025.1
mindmelting · Apr 5, 2024 · a0d18a4 · a0d18a4
1 parent 88d90a7
commit a0d18a4
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/custom_components/powerpal/sensor.py b/custom_components/powerpal/sensor.py
@@ -2,8 +2,7 @@
 from homeassistant.components.sensor import (
     SensorEntity,
     SensorDeviceClass,
-    STATE_CLASS_TOTAL_INCREASING,
-    STATE_CLASS_MEASUREMENT,
+    SensorStateClass,
 )
 from homeassistant.const import (
     UnitOfPower,
@@ -74,7 +73,7 @@ def device_class(self) -> str:
     @property
     def state_class(self) -> str:
         """Return the state class."""
-        return STATE_CLASS_TOTAL_INCREASING
+        return SensorStateClass.TOTAL_INCREASING
 
     @property
     def native_unit_of_measurement(self) -> str:
@@ -106,7 +105,7 @@ def device_class(self) -> str:
     @property
     def state_class(self) -> str:
         """Return the state class."""
-        return STATE_CLASS_MEASUREMENT
+        return SensorStateClass.MEASUREMENT
 
     @property
     def native_unit_of_measurement(self) -> str: