Update coordinator.py

cyr-ius · Nov 30, 2024 · a21b0f5 · a21b0f5
1 parent 931f8f8
commit a21b0f5
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/custom_components/heatzy/coordinator.py b/custom_components/heatzy/coordinator.py
@@ -16,7 +16,6 @@
 from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
 
 from .const import CONF_WEBSOCKET, DOMAIN
-from .fakedevices import FAKE_DEVICES
 
 _LOGGER = logging.getLogger(__name__)
 SCAN_INTERVAL = 60
@@ -92,8 +91,7 @@ async def _async_update_data(self) -> dict[str, Any]:
 
         try:
             if not self.api.websocket.is_updated:
-                # return await self.api.async_get_devices()
-                return FAKE_DEVICES
+                return await self.api.async_get_devices()
         except HeatzyException as error:
             raise UpdateFailed(f"Invalid response from API: {error}") from error
         else: