import ATTR_STATE_CLASS instead of CONF_STATE_CLASS

golles · Feb 4, 2022 · 47d4a35 · 47d4a35
1 parent 0c88de0
commit 47d4a35
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/custom_components/knmi/const.py b/custom_components/knmi/const.py
@@ -7,8 +7,10 @@
 from homeassistant.components.binary_sensor import (
     DEVICE_CLASS_SAFETY,
 )
-from homeassistant.components.sensor import STATE_CLASS_MEASUREMENT
-from homeassistant.components.sensor.const import CONF_STATE_CLASS
+from homeassistant.components.sensor import (
+    STATE_CLASS_MEASUREMENT,
+    ATTR_STATE_CLASS,
+)
 from homeassistant.components.weather import (
     ATTR_CONDITION_CLEAR_NIGHT,
     ATTR_CONDITION_CLOUDY,
@@ -71,7 +73,7 @@
         "device_class": DEVICE_CLASS_TEMPERATURE,
         "attributes": [
             {
-                "name": CONF_STATE_CLASS,
+                "name": ATTR_STATE_CLASS,
                 "value": STATE_CLASS_MEASUREMENT,
             },
         ],
@@ -84,7 +86,7 @@
         "device_class": DEVICE_CLASS_TEMPERATURE,
         "attributes": [
             {
-                "name": CONF_STATE_CLASS,
+                "name": ATTR_STATE_CLASS,
                 "value": STATE_CLASS_MEASUREMENT,
             },
         ],