Remove sleep and replace to refresh

cyr-ius · Oct 27, 2023 · 6dbae2e · 6dbae2e
1 parent 1370f47
commit 6dbae2e
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/custom_components/heatzy/climate.py b/custom_components/heatzy/climate.py
@@ -1,7 +1,6 @@
 """Climate sensors for Heatzy."""
 from __future__ import annotations
 
-import asyncio
 import logging
 from typing import Any
 
@@ -237,7 +236,7 @@ async def async_turn_on(self) -> None:
                         }
                     },
                 )
-                await asyncio.sleep(5)
+                await self.coordinator.async_request_refresh()
             await self.coordinator.api.async_control_device(
                 self.unique_id,
                 {CONF_ATTRS: {CONF_MODE: self.HA_TO_HEATZY_STATE[PRESET_COMFORT]}},
@@ -263,7 +262,7 @@ async def async_turn_off(self) -> None:
                         }
                     },
                 )
-                await asyncio.sleep(5)
+                await self.coordinator.async_request_refresh()
             await self.coordinator.api.async_control_device(
                 self.unique_id,
                 {CONF_ATTRS: {CONF_MODE: self.HEATZY_STOP}},