From b5de2c38b378696420313c2fc79843b717e77ebe Mon Sep 17 00:00:00 2001 From: epenet <6771947+epenet@users.noreply.github.com> Date: Mon, 20 Dec 2021 13:54:10 +0100 Subject: [PATCH] Use new enums in sms (#62372) Co-authored-by: epenet --- homeassistant/components/sms/sensor.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/homeassistant/components/sms/sensor.py b/homeassistant/components/sms/sensor.py index f5152ac7462..c51d1ef8773 100644 --- a/homeassistant/components/sms/sensor.py +++ b/homeassistant/components/sms/sensor.py @@ -3,8 +3,12 @@ import logging import gammu # pylint: disable=import-error -from homeassistant.components.sensor import SensorEntity, SensorEntityDescription -from homeassistant.const import DEVICE_CLASS_SIGNAL_STRENGTH, SIGNAL_STRENGTH_DECIBELS +from homeassistant.components.sensor import ( + SensorDeviceClass, + SensorEntity, + SensorEntityDescription, +) +from homeassistant.const import SIGNAL_STRENGTH_DECIBELS from homeassistant.helpers.entity import DeviceInfo from .const import DOMAIN, SMS_GATEWAY @@ -25,7 +29,7 @@ async def async_setup_entry(hass, config_entry, async_add_entities): SensorEntityDescription( key="signal", name=f"gsm_signal_imei_{imei}", - device_class=DEVICE_CLASS_SIGNAL_STRENGTH, + device_class=SensorDeviceClass.SIGNAL_STRENGTH, native_unit_of_measurement=SIGNAL_STRENGTH_DECIBELS, entity_registry_enabled_default=False, ),