Fix STATE_CLASS_MEASUREMENT

MiguelAngelLV · Dec 18, 2024 · 350aca4 · 350aca4
1 parent 40cacec
commit 350aca4
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/custom_components/octopus_spain/sensor.py b/custom_components/octopus_spain/sensor.py
@@ -14,8 +14,7 @@
 )
 
 from homeassistant.components.sensor import (
-    STATE_CLASS_MEASUREMENT,
-    SensorEntityDescription, SensorEntity
+    SensorEntityDescription, SensorEntity, SensorStateClass
 )
 from homeassistant.config_entries import ConfigEntry
 from homeassistant.core import HomeAssistant, callback
@@ -73,7 +72,7 @@ def __init__(self, account: str, key: str, name: str, coordinator, single: bool)
             key=f"{key}_{account}",
             icon="mdi:piggy-bank-outline",
             native_unit_of_measurement=CURRENCY_EURO,
-            state_class=STATE_CLASS_MEASUREMENT
+            state_class=SensorStateClass.MEASUREMENT
         )
 
     async def async_added_to_hass(self) -> None:
@@ -104,7 +103,7 @@ def __init__(self, account: str, coordinator, single: bool):
             key=f"last_invoice_{account}",
             icon="mdi:currency-eur",
             native_unit_of_measurement=CURRENCY_EURO,
-            state_class=STATE_CLASS_MEASUREMENT
+            state_class=SensorStateClass.MEASUREMENT
         )
 
     async def async_added_to_hass(self) -> None: