From 7c8d2353567ccc3529c8b839d05668f4acb96746 Mon Sep 17 00:00:00 2001
From: epenet <6771947+epenet@users.noreply.github.com>
Date: Thu, 16 Dec 2021 15:51:36 +0100
Subject: [PATCH] Use SensorDeviceClass in mfi (#61900)

Co-authored-by: epenet <epenet@users.noreply.github.com>
---
 homeassistant/components/mfi/sensor.py | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/homeassistant/components/mfi/sensor.py b/homeassistant/components/mfi/sensor.py
index b27f719d974..a951f10822f 100644
--- a/homeassistant/components/mfi/sensor.py
+++ b/homeassistant/components/mfi/sensor.py
@@ -5,7 +5,11 @@ from mficlient.client import FailedToLogin, MFiClient
 import requests
 import voluptuous as vol
 
-from homeassistant.components.sensor import PLATFORM_SCHEMA, SensorEntity
+from homeassistant.components.sensor import (
+    PLATFORM_SCHEMA,
+    SensorDeviceClass,
+    SensorEntity,
+)
 from homeassistant.const import (
     CONF_HOST,
     CONF_PASSWORD,
@@ -13,7 +17,6 @@ from homeassistant.const import (
     CONF_SSL,
     CONF_USERNAME,
     CONF_VERIFY_SSL,
-    DEVICE_CLASS_TEMPERATURE,
     STATE_OFF,
     STATE_ON,
     TEMP_CELSIUS,
@@ -110,7 +113,7 @@ class MfiSensor(SensorEntity):
             return None
 
         if tag == "temperature":
-            return DEVICE_CLASS_TEMPERATURE
+            return SensorDeviceClass.TEMPERATURE
 
         return None