From 59e824eaede0e07964f284c44d29cbfe2960feed Mon Sep 17 00:00:00 2001 From: cyr-ius Date: Thu, 4 Jan 2024 12:17:00 +0100 Subject: [PATCH] Fix config_flow error --- custom_components/heatzy/climate.py | 2 +- custom_components/heatzy/config_flow.py | 8 ++++++-- custom_components/heatzy/switch.py | 2 +- 3 files changed, 8 insertions(+), 4 deletions(-) diff --git a/custom_components/heatzy/climate.py b/custom_components/heatzy/climate.py index 653e413..46d3169 100644 --- a/custom_components/heatzy/climate.py +++ b/custom_components/heatzy/climate.py @@ -4,8 +4,8 @@ import logging from typing import Any -from heatzypy.exception import HeatzyException import voluptuous as vol +from wsheatzypy.exception import HeatzyException from homeassistant.components.climate import ( ATTR_TARGET_TEMP_HIGH, diff --git a/custom_components/heatzy/config_flow.py b/custom_components/heatzy/config_flow.py index 062f2e3..b920f46 100644 --- a/custom_components/heatzy/config_flow.py +++ b/custom_components/heatzy/config_flow.py @@ -3,9 +3,13 @@ import logging -from heatzypy import HeatzyClient -from heatzypy.exception import AuthenticationFailed, HeatzyException, HttpRequestFailed import voluptuous as vol +from wsheatzypy import HeatzyClient +from wsheatzypy.exception import ( + AuthenticationFailed, + HeatzyException, + HttpRequestFailed, +) from homeassistant import config_entries from homeassistant.const import CONF_PASSWORD, CONF_USERNAME diff --git a/custom_components/heatzy/switch.py b/custom_components/heatzy/switch.py index 1ba2105..968bc01 100644 --- a/custom_components/heatzy/switch.py +++ b/custom_components/heatzy/switch.py @@ -3,7 +3,7 @@ import logging -from heatzypy.exception import HeatzyException +from wsheatzypy.exception import HeatzyException from homeassistant.components.switch import SwitchEntity from homeassistant.config_entries import ConfigEntry