Use climate enums in flexit (#70654)

This commit is contained in:
epenet 2022-04-25 10:47:12 +02:00 committed by GitHub
parent 587505c85b
commit 8db4994345
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -5,12 +5,8 @@ import logging
import voluptuous as vol import voluptuous as vol
from homeassistant.components.climate import ( from homeassistant.components.climate import PLATFORM_SCHEMA, ClimateEntity
PLATFORM_SCHEMA, from homeassistant.components.climate.const import ClimateEntityFeature, HVACMode
ClimateEntity,
ClimateEntityFeature,
)
from homeassistant.components.climate.const import HVAC_MODE_COOL
from homeassistant.components.modbus import get_hub from homeassistant.components.modbus import get_hub
from homeassistant.components.modbus.const import ( from homeassistant.components.modbus.const import (
CALL_TYPE_REGISTER_HOLDING, CALL_TYPE_REGISTER_HOLDING,
@ -187,7 +183,7 @@ class Flexit(ClimateEntity):
Need to be a subset of HVAC_MODES. Need to be a subset of HVAC_MODES.
""" """
return [HVAC_MODE_COOL] return [HVACMode.COOL]
@property @property
def fan_mode(self): def fan_mode(self):