#111 update enums

elden1337 · May 4, 2024 · feeee67 · feeee67
1 parent 6df790b
commit feeee67
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/custom_components/peaqhvac/climate.py b/custom_components/peaqhvac/climate.py
@@ -4,9 +4,7 @@
 from homeassistant.components.climate import ClimateEntity
 from homeassistant.components.climate.const import (PRESET_AWAY, PRESET_ECO,
                                                     PRESET_NONE,
-                                                    SUPPORT_PRESET_MODE,
-                                                    SUPPORT_TARGET_TEMPERATURE,
-                                                    HVACAction, HVACMode)
+                                                    HVACAction, HVACMode, ClimateEntityFeature)
 from homeassistant.const import ATTR_TEMPERATURE, TEMP_CELSIUS
 from homeassistant.core import HomeAssistant
 from homeassistant.helpers.restore_state import RestoreEntity
@@ -46,7 +44,7 @@ def __init__(self, hass, entry_id, hub, name):
 
     @property
     def supported_features(self):
-        return SUPPORT_TARGET_TEMPERATURE | SUPPORT_PRESET_MODE
+        return ClimateEntityFeature.TARGET_TEMPERATURE | ClimateEntityFeature.PRESET_MODE
 
     @property
     def name(self):