diff --git a/homeassistant/components/flexit/climate.py b/homeassistant/components/flexit/climate.py index dbe1e060a12..b833617f2ca 100644 --- a/homeassistant/components/flexit/climate.py +++ b/homeassistant/components/flexit/climate.py @@ -16,8 +16,6 @@ from homeassistant.components.climate import ( from homeassistant.components.modbus import ( CALL_TYPE_REGISTER_HOLDING, CALL_TYPE_REGISTER_INPUT, - CALL_TYPE_WRITE_REGISTER, - CONF_HUB, DEFAULT_HUB, ModbusHub, get_hub, @@ -34,6 +32,9 @@ import homeassistant.helpers.config_validation as cv from homeassistant.helpers.entity_platform import AddEntitiesCallback from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType +CALL_TYPE_WRITE_REGISTER = "write_register" +CONF_HUB = "hub" + PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend( { vol.Optional(CONF_HUB, default=DEFAULT_HUB): cv.string,