From f512bacfc7a6db5fae6b11f8c63132112d793c16 Mon Sep 17 00:00:00 2001 From: epenet <6771947+epenet@users.noreply.github.com> Date: Thu, 9 Dec 2021 22:20:06 +0100 Subject: [PATCH] Use new SensorDeviceClass enum in emonitor (#61385) Co-authored-by: epenet --- homeassistant/components/emonitor/sensor.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/homeassistant/components/emonitor/sensor.py b/homeassistant/components/emonitor/sensor.py index 5d4f1983ac1..a0b0e3c1fd9 100644 --- a/homeassistant/components/emonitor/sensor.py +++ b/homeassistant/components/emonitor/sensor.py @@ -2,7 +2,7 @@ from aioemonitor.monitor import EmonitorChannel -from homeassistant.components.sensor import DEVICE_CLASS_POWER, SensorEntity +from homeassistant.components.sensor import SensorDeviceClass, SensorEntity from homeassistant.const import POWER_WATT from homeassistant.helpers import device_registry as dr from homeassistant.helpers.entity import DeviceInfo @@ -37,7 +37,7 @@ async def async_setup_entry(hass, config_entry, async_add_entities): class EmonitorPowerSensor(CoordinatorEntity, SensorEntity): """Representation of an Emonitor power sensor entity.""" - _attr_device_class = DEVICE_CLASS_POWER + _attr_device_class = SensorDeviceClass.POWER _attr_native_unit_of_measurement = POWER_WATT def __init__(self, coordinator: DataUpdateCoordinator, channel_number: int) -> None: