Generics and other type hint improvements (#45250)

homeassistant/helpers/entity_values.py

@@ -6,18 +6,20 @@ from typing import Any, Dict, Optional, Pattern
 
 from homeassistant.core import split_entity_id
 
+# mypy: disallow-any-generics
+
 
 class EntityValues:
     """Class to store entity id based values."""
 
     def __init__(
         self,
-        exact: Optional[Dict] = None,
-        domain: Optional[Dict] = None,
-        glob: Optional[Dict] = None,
+        exact: Optional[Dict[str, Dict[str, str]]] = None,
+        domain: Optional[Dict[str, Dict[str, str]]] = None,
+        glob: Optional[Dict[str, Dict[str, str]]] = None,
     ) -> None:
         """Initialize an EntityConfigDict."""
-        self._cache: Dict[str, Dict] = {}
+        self._cache: Dict[str, Dict[str, str]] = {}
         self._exact = exact
         self._domain = domain
 
@@ -30,7 +32,7 @@ class EntityValues:
 
         self._glob = compiled
 
-    def get(self, entity_id: str) -> Dict:
+    def get(self, entity_id: str) -> Dict[str, str]:
         """Get config for an entity id."""
         if entity_id in self._cache:
             return self._cache[entity_id]