From fc6c66fe6c5b48c669b50f45869a2670b5c5087e Mon Sep 17 00:00:00 2001
From: epenet <6771947+epenet@users.noreply.github.com>
Date: Wed, 17 Aug 2022 18:42:35 +0200
Subject: [PATCH] Add RestoreEntity to button checks in pylint (#76932)

---
 pylint/plugins/hass_enforce_type_hints.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/pylint/plugins/hass_enforce_type_hints.py b/pylint/plugins/hass_enforce_type_hints.py
index 004f01aab42..4e81b7b77af 100644
--- 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=[