From 9d66dd35ccf2459e5d0b840c1d0a8f1d304144b9 Mon Sep 17 00:00:00 2001
From: Robert Hillis <tkdrob4390@yahoo.com>
Date: Thu, 16 Dec 2021 03:21:58 -0500
Subject: [PATCH] Use enums in waqi (#61998)

---
 homeassistant/components/waqi/sensor.py | 11 +++++++----
 1 file changed, 7 insertions(+), 4 deletions(-)

diff --git a/homeassistant/components/waqi/sensor.py b/homeassistant/components/waqi/sensor.py
index 98ee8db4e89..331e689f14a 100644
--- a/homeassistant/components/waqi/sensor.py
+++ b/homeassistant/components/waqi/sensor.py
@@ -7,13 +7,16 @@ import aiohttp
 import voluptuous as vol
 from waqiasync import WaqiClient
 
-from homeassistant.components.sensor import STATE_CLASS_MEASUREMENT, SensorEntity
+from homeassistant.components.sensor import (
+    SensorDeviceClass,
+    SensorEntity,
+    SensorStateClass,
+)
 from homeassistant.const import (
     ATTR_ATTRIBUTION,
     ATTR_TEMPERATURE,
     ATTR_TIME,
     CONF_TOKEN,
-    DEVICE_CLASS_AQI,
 )
 from homeassistant.exceptions import PlatformNotReady
 from homeassistant.helpers.aiohttp_client import async_get_clientsession
@@ -102,8 +105,8 @@ class WaqiSensor(SensorEntity):
 
     _attr_icon = ATTR_ICON
     _attr_native_unit_of_measurement = ATTR_UNIT
-    _attr_device_class = DEVICE_CLASS_AQI
-    _attr_state_class = STATE_CLASS_MEASUREMENT
+    _attr_device_class = SensorDeviceClass.AQI
+    _attr_state_class = SensorStateClass.MEASUREMENT
 
     def __init__(self, client, station):
         """Initialize the sensor."""