From 8c39eade5e46bfdea9a28372fa53a4865c62bea0 Mon Sep 17 00:00:00 2001
From: epenet <6771947+epenet@users.noreply.github.com>
Date: Fri, 10 Dec 2021 09:14:06 +0100
Subject: [PATCH] Use SensorDeviceClass in environment_canada (#61390)

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

diff --git a/homeassistant/components/environment_canada/sensor.py b/homeassistant/components/environment_canada/sensor.py
index 0d33a166254..e71411913c9 100644
--- a/homeassistant/components/environment_canada/sensor.py
+++ b/homeassistant/components/environment_canada/sensor.py
@@ -4,12 +4,15 @@ import re
 
 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 (
     ATTR_LOCATION,
     CONF_LATITUDE,
     CONF_LONGITUDE,
-    DEVICE_CLASS_TEMPERATURE,
     TEMP_CELSIUS,
 )
 import homeassistant.helpers.config_validation as cv
@@ -135,7 +138,7 @@ class ECSensor(CoordinatorEntity, SensorEntity):
             "humidex",
         ):
             self._unit = TEMP_CELSIUS
-            self._device_class = DEVICE_CLASS_TEMPERATURE
+            self._device_class = SensorDeviceClass.TEMPERATURE
         else:
             self._unit = sensor_data.get("unit")