Merge pull request #37 from rccoleman/dev

Migrate away from deprecated DEVICE_CLASS_*
rccoleman · Jan 5, 2023 · ca1fc7f · ca1fc7f
2 parents 9f4ff98 + 6a57c53
commit ca1fc7f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/custom_components/lamarzocco/binary_sensor.py b/custom_components/lamarzocco/binary_sensor.py
@@ -3,7 +3,7 @@
 import logging
 
 from homeassistant.components.binary_sensor import (
-    DEVICE_CLASS_PROBLEM,
+    BinarySensorDeviceClass,
     BinarySensorEntity,
 )
 
@@ -38,7 +38,7 @@
         },
         ENTITY_TYPE: TYPE_WATER_RESERVOIR_CONTACT,
         ENTITY_ICON: "mdi:water-well",
-        ENTITY_CLASS: DEVICE_CLASS_PROBLEM,
+        ENTITY_CLASS: BinarySensorDeviceClass.PROBLEM,
     }
 }