Use enums in wolflink (#61988)

pull/62011/head
Robert Hillis 2021-12-16 03:10:14 -05:00 committed by GitHub
parent f691b0a1da
commit 806366a0c1
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 10 deletions

View File

@ -9,14 +9,8 @@ from wolf_smartset.models import (
Temperature,
)
from homeassistant.components.sensor import SensorEntity
from homeassistant.const import (
DEVICE_CLASS_PRESSURE,
DEVICE_CLASS_TEMPERATURE,
PRESSURE_BAR,
TEMP_CELSIUS,
TIME_HOURS,
)
from homeassistant.components.sensor import SensorDeviceClass, SensorEntity
from homeassistant.const import PRESSURE_BAR, TEMP_CELSIUS, TIME_HOURS
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .const import COORDINATOR, DEVICE_ID, DOMAIN, PARAMETERS, STATES
@ -106,7 +100,7 @@ class WolfLinkTemperature(WolfLinkSensor):
@property
def device_class(self):
"""Return the device_class."""
return DEVICE_CLASS_TEMPERATURE
return SensorDeviceClass.TEMPERATURE
@property
def native_unit_of_measurement(self):
@ -120,7 +114,7 @@ class WolfLinkPressure(WolfLinkSensor):
@property
def device_class(self):
"""Return the device_class."""
return DEVICE_CLASS_PRESSURE
return SensorDeviceClass.PRESSURE
@property
def native_unit_of_measurement(self):