From e049a857853eb310693d4c6fa5f25b5ec7d73d58 Mon Sep 17 00:00:00 2001 From: G Johansson Date: Sun, 3 May 2026 17:36:45 +0000 Subject: [PATCH] More mods --- homeassistant/components/compensation/__init__.py | 5 ++++- homeassistant/components/compensation/config_flow.py | 2 +- homeassistant/components/compensation/const.py | 6 +++++- 3 files changed, 10 insertions(+), 3 deletions(-) diff --git a/homeassistant/components/compensation/__init__.py b/homeassistant/components/compensation/__init__.py index 7ab042ab5b30..54890de1605f 100644 --- a/homeassistant/components/compensation/__init__.py +++ b/homeassistant/components/compensation/__init__.py @@ -187,4 +187,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: """Unload Compensation config entry.""" - return await hass.config_entries.async_unload_platforms(entry, PLATFORMS) + unload_ok = await hass.config_entries.async_unload_platforms(entry, PLATFORMS) + if unload_ok: + hass.data[DATA_COMPENSATION].pop(entry.entry_id, None) + return unload_ok diff --git a/homeassistant/components/compensation/config_flow.py b/homeassistant/components/compensation/config_flow.py index 23e421e3b6cb..a37d810dda74 100644 --- a/homeassistant/components/compensation/config_flow.py +++ b/homeassistant/components/compensation/config_flow.py @@ -1,4 +1,4 @@ -"""Config flow for statistics.""" +"""Config flow for Compensation integration.""" from __future__ import annotations diff --git a/homeassistant/components/compensation/const.py b/homeassistant/components/compensation/const.py index 6b8f0c7f1d0a..7e08202ba437 100644 --- a/homeassistant/components/compensation/const.py +++ b/homeassistant/components/compensation/const.py @@ -1,6 +1,9 @@ """Compensation constants.""" +from typing import Any + from homeassistant.const import Platform +from homeassistant.util.hass_dict import HassKey DOMAIN = "compensation" PLATFORMS = [Platform.SENSOR] @@ -16,7 +19,8 @@ CONF_PRECISION = "precision" CONF_POLYNOMIAL = "polynomial" CONF_POLYNOMIAL_CONFIG = "polynomial_config" -DATA_COMPENSATION = "compensation_data" + +DATA_COMPENSATION: HassKey[dict[str, Any]] = HassKey("compensation_data") DEFAULT_DEGREE = 1 DEFAULT_NAME = "Compensation"