Fix lint

custom_components/heatzy/climate.py

@@ -5,21 +5,19 @@
 from heatzypy.exception import HeatzyException
 
 from homeassistant.components.climate import (
-    ClimateEntity,
-    ClimateEntityFeature,
-    HVACAction,
-    HVACMode,
-)
-from homeassistant.components.climate.const import (
     ATTR_TARGET_TEMP_HIGH,
     ATTR_TARGET_TEMP_LOW,
     PRESET_AWAY,
     PRESET_COMFORT,
     PRESET_ECO,
     PRESET_NONE,
+    ClimateEntity,
+    ClimateEntityFeature,
+    HVACAction,
+    HVACMode,
 )
 from homeassistant.config_entries import ConfigEntry
-from homeassistant.const import TEMP_CELSIUS
+from homeassistant.const import UnitOfTemperature
 from homeassistant.core import HomeAssistant
 from homeassistant.helpers.entity import DeviceInfo
 from homeassistant.helpers.entity_platform import AddEntitiesCallback
@@ -79,10 +77,10 @@ class HeatzyThermostat(CoordinatorEntity[HeatzyDataUpdateCoordinator], ClimateEn
     _attr_hvac_modes = [HVACMode.HEAT, HVACMode.OFF, HVACMode.AUTO]
     _attr_preset_modes = [PRESET_COMFORT, PRESET_ECO, PRESET_AWAY]
     _attr_supported_features = ClimateEntityFeature.PRESET_MODE
-    _attr_temperature_unit = TEMP_CELSIUS
+    _attr_temperature_unit = UnitOfTemperature.CELSIUS
     _attr_has_entity_name = True
 
-    def __init__(self, coordinator: HeatzyDataUpdateCoordinator, unique_id):
+    def __init__(self, coordinator: HeatzyDataUpdateCoordinator, unique_id) -> None:
         """Init."""
         super().__init__(coordinator)
         self._attr_unique_id = unique_id

custom_components/heatzy/config_flow.py

@@ -4,6 +4,7 @@
 import voluptuous as vol
 from heatzypy import HeatzyClient
 from heatzypy.exception import AuthenticationFailed, HeatzyException, HttpRequestFailed
+
 from homeassistant import config_entries
 from homeassistant.const import CONF_PASSWORD, CONF_USERNAME
 from homeassistant.helpers.aiohttp_client import async_create_clientsession

custom_components/heatzy/coordinator.py

@@ -5,6 +5,7 @@
 import async_timeout
 from heatzypy import HeatzyClient
 from heatzypy.exception import AuthenticationFailed, HeatzyException
+
 from homeassistant.config_entries import ConfigEntry
 from homeassistant.const import CONF_PASSWORD, CONF_USERNAME
 from homeassistant.core import HomeAssistant

custom_components/heatzy/diagnostics.py

@@ -1,6 +1,8 @@
 """Diagnostics support for Heatzy."""
 from __future__ import annotations
 
+from typing import Any
+
 from homeassistant.components.diagnostics import async_redact_data
 from homeassistant.config_entries import ConfigEntry
 from homeassistant.core import HomeAssistant

custom_components/heatzy/manifest.json

@@ -9,5 +9,5 @@
     "issue_tracker": "https://github.com/cyr-ius/hass-heatzy/issues",
     "loggers": ["heatzypy"],
     "requirements": ["heatzypy==2.1.5"],
-    "version": "5.8.3"
+    "version": "5.8.4"
 }