From d2216363c5c69f59135f58065de6aee25758ea8c Mon Sep 17 00:00:00 2001 From: epenet <6771947+epenet@users.noreply.github.com> Date: Wed, 8 Dec 2021 21:57:25 +0100 Subject: [PATCH] Use new DeviceClass and StateClass enums in aseko_pool_live (#61276) Co-authored-by: epenet --- homeassistant/components/aseko_pool_live/sensor.py | 13 ++++++++----- 1 file changed, 8 insertions(+), 5 deletions(-) diff --git a/homeassistant/components/aseko_pool_live/sensor.py b/homeassistant/components/aseko_pool_live/sensor.py index 41036b582a7..74051ef454f 100644 --- a/homeassistant/components/aseko_pool_live/sensor.py +++ b/homeassistant/components/aseko_pool_live/sensor.py @@ -3,9 +3,12 @@ from __future__ import annotations from aioaseko import Unit, Variable -from homeassistant.components.sensor import STATE_CLASS_MEASUREMENT, SensorEntity +from homeassistant.components.sensor import ( + SensorDeviceClass, + SensorEntity, + SensorStateClass, +) from homeassistant.config_entries import ConfigEntry -from homeassistant.const import DEVICE_CLASS_TEMPERATURE from homeassistant.core import HomeAssistant from homeassistant.helpers.entity_platform import AddEntitiesCallback @@ -33,7 +36,7 @@ async def async_setup_entry( class VariableSensorEntity(AsekoEntity, SensorEntity): """Representation of a unit variable sensor entity.""" - attr_state_class = STATE_CLASS_MEASUREMENT + attr_state_class = SensorStateClass.MEASUREMENT def __init__( self, unit: Unit, variable: Variable, coordinator: AsekoDataUpdateCoordinator @@ -61,8 +64,8 @@ class VariableSensorEntity(AsekoEntity, SensorEntity): }.get(self._variable.type) self._attr_device_class = { - "airTemp": DEVICE_CLASS_TEMPERATURE, - "waterTemp": DEVICE_CLASS_TEMPERATURE, + "airTemp": SensorDeviceClass.TEMPERATURE, + "waterTemp": SensorDeviceClass.TEMPERATURE, }.get(self._variable.type) @property