Use new device_class tuple

custom_components/somneo/manifest.json

@@ -1,7 +1,7 @@
 {
     "domain": "somneo",
     "name": "Philips Somneo",
-    "version": "2.2.1",
+    "version": "2.2.2",
     "documentation": "https://github.com/theneweinstein/somneo",
     "issue_tracker": "https://github.com/theneweinstein/somneo/issues",
     "dependencies": [],

custom_components/somneo/sensor.py

@@ -3,8 +3,7 @@
 
 from datetime import datetime
 
-from homeassistant.const import DEVICE_CLASS_HUMIDITY, DEVICE_CLASS_ILLUMINANCE, DEVICE_CLASS_TEMPERATURE, DEVICE_CLASS_PRESSURE, DEVICE_CLASS_TIMESTAMP
-from homeassistant.components.sensor import STATE_CLASS_MEASUREMENT, SensorEntity
+from homeassistant.components.sensor import SensorDeviceClass, STATE_CLASS_MEASUREMENT, SensorEntity
 
 from .const import *
 
@@ -47,16 +46,16 @@ def __init__(self, name, data, device_info, serial, sensor_type):
         self._attr_unique_id = serial + '_' + sensor_type
 
     @property
-    def device_class(self):
+    def device_class(self) -> SensorDeviceClass:
         """Return the class of this device, from component DEVICE_CLASSES."""
         if self._type == "temperature":
-            return DEVICE_CLASS_TEMPERATURE
+            return SensorDeviceClass.TEMPERATURE
         if self._type == "humidity":
-            return DEVICE_CLASS_HUMIDITY
+            return SensorDeviceClass.HUMIDITY
         if self._type == "luminance":
-            return DEVICE_CLASS_ILLUMINANCE
+            return SensorDeviceClass.ILLUMINANCE
         if self._type == "pressure":
-            return DEVICE_CLASS_PRESSURE
+            return SensorDeviceClass.PRESSURE
         else:
             return None
 
@@ -82,7 +81,7 @@ def __init__(self, name, data, device_info, serial):
         self._attr_device_info = device_info
         self._attr_unique_id = serial + '_next_alarm'
         self._attr_native_value = None
-        self._attr_device_class = DEVICE_CLASS_TIMESTAMP
+        self._attr_device_class = SensorDeviceClass.TIMESTAMP
 
     async def async_update(self):
         """Get the latest data and updates the states."""