From a34bce6202fe67da3558ef3a46942dbf73f1fa29 Mon Sep 17 00:00:00 2001
From: epenet <6771947+epenet@users.noreply.github.com>
Date: Mon, 19 May 2025 10:59:46 +0200
Subject: [PATCH] Fix runtime_data in iqvia (#145181)

---
 homeassistant/components/iqvia/entity.py | 8 ++++----
 homeassistant/components/iqvia/sensor.py | 7 ++-----
 2 files changed, 6 insertions(+), 9 deletions(-)

diff --git a/homeassistant/components/iqvia/entity.py b/homeassistant/components/iqvia/entity.py
index 1964a7cb039..04e92ef9c4d 100644
--- a/homeassistant/components/iqvia/entity.py
+++ b/homeassistant/components/iqvia/entity.py
@@ -6,7 +6,7 @@ from homeassistant.core import callback
 from homeassistant.helpers.entity import EntityDescription
 from homeassistant.helpers.update_coordinator import CoordinatorEntity
 
-from .const import CONF_ZIP_CODE, DOMAIN, TYPE_ALLERGY_FORECAST, TYPE_ALLERGY_OUTLOOK
+from .const import CONF_ZIP_CODE, TYPE_ALLERGY_FORECAST, TYPE_ALLERGY_OUTLOOK
 from .coordinator import IqviaConfigEntry, IqviaUpdateCoordinator
 
 
@@ -44,9 +44,9 @@ class IQVIAEntity(CoordinatorEntity[IqviaUpdateCoordinator]):
 
         if self.entity_description.key == TYPE_ALLERGY_FORECAST:
             self.async_on_remove(
-                self.hass.data[DOMAIN][self._entry.entry_id][
-                    TYPE_ALLERGY_OUTLOOK
-                ].async_add_listener(self._handle_coordinator_update)
+                self._entry.runtime_data[TYPE_ALLERGY_OUTLOOK].async_add_listener(
+                    self._handle_coordinator_update
+                )
             )
 
         self.update_from_latest_data()
diff --git a/homeassistant/components/iqvia/sensor.py b/homeassistant/components/iqvia/sensor.py
index c0401b27368..8b838d35ea1 100644
--- a/homeassistant/components/iqvia/sensor.py
+++ b/homeassistant/components/iqvia/sensor.py
@@ -17,7 +17,6 @@ from homeassistant.core import HomeAssistant, callback
 from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
 
 from .const import (
-    DOMAIN,
     TYPE_ALLERGY_FORECAST,
     TYPE_ALLERGY_INDEX,
     TYPE_ALLERGY_OUTLOOK,
@@ -145,7 +144,7 @@ async def async_setup_entry(
     sensors.extend(
         [
             IndexSensor(
-                hass.data[DOMAIN][entry.entry_id][
+                entry.runtime_data[
                     API_CATEGORY_MAPPING.get(description.key, description.key)
                 ],
                 entry,
@@ -207,9 +206,7 @@ class ForecastSensor(IQVIAEntity, SensorEntity):
         )
 
         if self.entity_description.key == TYPE_ALLERGY_FORECAST:
-            outlook_coordinator = self.hass.data[DOMAIN][self._entry.entry_id][
-                TYPE_ALLERGY_OUTLOOK
-            ]
+            outlook_coordinator = self._entry.runtime_data[TYPE_ALLERGY_OUTLOOK]
 
             if not outlook_coordinator.last_update_success:
                 return