From 502231b7d2b9f4add7b558cc95cfc45ce99f9bd2 Mon Sep 17 00:00:00 2001 From: Jan-Philipp Benecke Date: Sat, 30 Mar 2024 21:15:52 +0100 Subject: [PATCH] Avoid call to `hass.helpers.store` in CategoryRegistry (#114485) --- homeassistant/helpers/category_registry.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/homeassistant/helpers/category_registry.py b/homeassistant/helpers/category_registry.py index ee0c8c1bb88077..fec872623744d9 100644 --- a/homeassistant/helpers/category_registry.py +++ b/homeassistant/helpers/category_registry.py @@ -11,6 +11,7 @@ from homeassistant.util.ulid import ulid_now from .registry import BaseRegistry +from .storage import Store from .typing import UNDEFINED, EventType, UndefinedType DATA_REGISTRY = "category_registry" @@ -46,7 +47,8 @@ def __init__(self, hass: HomeAssistant) -> None: """Initialize the category registry.""" self.hass = hass self.categories: dict[str, dict[str, CategoryEntry]] = {} - self._store = hass.helpers.storage.Store( + self._store: Store[dict[str, dict[str, list[dict[str, str]]]]] = Store( + hass, STORAGE_VERSION_MAJOR, STORAGE_KEY, atomic_writes=True,