From 00b90dff5546f8a4c114b0d540af352bfbef6b5c Mon Sep 17 00:00:00 2001
From: epenet <6771947+epenet@users.noreply.github.com>
Date: Mon, 13 Dec 2021 17:36:51 +0100
Subject: [PATCH] Use SensorDeviceClass in greeneye-monitor (#61653)

Co-authored-by: epenet <epenet@users.noreply.github.com>
---
 homeassistant/components/greeneye_monitor/sensor.py | 11 ++++-------
 1 file changed, 4 insertions(+), 7 deletions(-)

diff --git a/homeassistant/components/greeneye_monitor/sensor.py b/homeassistant/components/greeneye_monitor/sensor.py
index 57c5c79891d..b0d1a246819 100644
--- a/homeassistant/components/greeneye_monitor/sensor.py
+++ b/homeassistant/components/greeneye_monitor/sensor.py
@@ -5,15 +5,12 @@ from typing import Any, Optional, Union, cast
 
 import greeneye
 
-from homeassistant.components.sensor import SensorEntity
+from homeassistant.components.sensor import SensorDeviceClass, SensorEntity
 from homeassistant.const import (
     CONF_NAME,
     CONF_SENSOR_TYPE,
     CONF_SENSORS,
     CONF_TEMPERATURE_UNIT,
-    DEVICE_CLASS_POWER,
-    DEVICE_CLASS_TEMPERATURE,
-    DEVICE_CLASS_VOLTAGE,
     ELECTRIC_POTENTIAL_VOLT,
     POWER_WATT,
     TIME_HOURS,
@@ -162,7 +159,7 @@ class CurrentSensor(GEMSensor):
     """Entity showing power usage on one channel of the monitor."""
 
     _attr_native_unit_of_measurement = UNIT_WATTS
-    _attr_device_class = DEVICE_CLASS_POWER
+    _attr_device_class = SensorDeviceClass.POWER
 
     def __init__(
         self, monitor_serial_number: int, number: int, name: str, net_metering: bool
@@ -261,7 +258,7 @@ class PulseCounter(GEMSensor):
 class TemperatureSensor(GEMSensor):
     """Entity showing temperature from one temperature sensor."""
 
-    _attr_device_class = DEVICE_CLASS_TEMPERATURE
+    _attr_device_class = SensorDeviceClass.TEMPERATURE
 
     def __init__(
         self, monitor_serial_number: int, number: int, name: str, unit: str
@@ -291,7 +288,7 @@ class VoltageSensor(GEMSensor):
     """Entity showing voltage."""
 
     _attr_native_unit_of_measurement = ELECTRIC_POTENTIAL_VOLT
-    _attr_device_class = DEVICE_CLASS_VOLTAGE
+    _attr_device_class = SensorDeviceClass.VOLTAGE
 
     def __init__(self, monitor_serial_number: int, number: int, name: str) -> None:
         """Construct the entity."""