Add RestoreEntity to button checks in pylint (#76932)

2022-08-17 18:42:35 +02:00 · 2022-08-17 18:42:35 +02:00 · fc6c66fe6c
parent a63a3b96b7
commit fc6c66fe6c
1 changed files with 4 additions and 0 deletions
--- a/pylint/plugins/hass_enforce_type_hints.py
+++ b/pylint/plugins/hass_enforce_type_hints.py
@ -724,6 +724,10 @@ _INHERITANCE_MATCH: dict[str, list[ClassTypeHintMatch]] = {
            base_class="Entity",
            matches=_ENTITY_MATCH,
        ),
+        ClassTypeHintMatch(
+            base_class="RestoreEntity",
+            matches=_RESTORE_ENTITY_MATCH,
+        ),
        ClassTypeHintMatch(
            base_class="ButtonEntity",
            matches=[