From ebf9faac17d14014acf5b7b70c96d3f763d3cf75 Mon Sep 17 00:00:00 2001 From: epenet <6771947+epenet@users.noreply.github.com> Date: Wed, 8 Dec 2021 22:33:09 +0100 Subject: [PATCH] Use new SensorDeviceClass enum in atag (#61278) Co-authored-by: epenet --- homeassistant/components/atag/sensor.py | 8 +++----- 1 file changed, 3 insertions(+), 5 deletions(-) diff --git a/homeassistant/components/atag/sensor.py b/homeassistant/components/atag/sensor.py index 386b5999712..0f1599098a6 100644 --- a/homeassistant/components/atag/sensor.py +++ b/homeassistant/components/atag/sensor.py @@ -1,8 +1,6 @@ """Initialization of ATAG One sensor platform.""" -from homeassistant.components.sensor import SensorEntity +from homeassistant.components.sensor import SensorDeviceClass, SensorEntity from homeassistant.const import ( - DEVICE_CLASS_PRESSURE, - DEVICE_CLASS_TEMPERATURE, PERCENTAGE, PRESSURE_BAR, TEMP_CELSIUS, @@ -38,8 +36,8 @@ class AtagSensor(AtagEntity, SensorEntity): super().__init__(coordinator, SENSORS[sensor]) self._attr_name = sensor if coordinator.data.report[self._id].sensorclass in ( - DEVICE_CLASS_PRESSURE, - DEVICE_CLASS_TEMPERATURE, + SensorDeviceClass.PRESSURE, + SensorDeviceClass.TEMPERATURE, ): self._attr_device_class = coordinator.data.report[self._id].sensorclass if coordinator.data.report[self._id].measure in (