diff --git a/homeassistant/components/atag/climate.py b/homeassistant/components/atag/climate.py index ce52bd4fd65..00897f127d9 100644 --- a/homeassistant/components/atag/climate.py +++ b/homeassistant/components/atag/climate.py @@ -15,6 +15,7 @@ from homeassistant.config_entries import ConfigEntry from homeassistant.const import ATTR_TEMPERATURE, Platform from homeassistant.core import HomeAssistant from homeassistant.helpers.entity_platform import AddEntitiesCallback +from homeassistant.util.enum import try_parse_enum from . import DOMAIN, AtagEntity @@ -52,11 +53,9 @@ class AtagThermostat(AtagEntity, ClimateEntity): self._attr_temperature_unit = coordinator.data.climate.temp_unit @property - def hvac_mode(self) -> str | None: + def hvac_mode(self) -> HVACMode | None: """Return hvac operation ie. heat, cool mode.""" - if self.coordinator.data.climate.hvac_mode in HVAC_MODES: - return self.coordinator.data.climate.hvac_mode - return None + return try_parse_enum(HVACMode, self.coordinator.data.climate.hvac_mode) @property def hvac_action(self) -> str | None: