From ff2f0ac32078ffac0f7ad43875ea173c7471834d Mon Sep 17 00:00:00 2001
From: epenet <6771947+epenet@users.noreply.github.com>
Date: Tue, 17 Feb 2026 15:34:16 +0100
Subject: [PATCH] Mark RestoreEntity/RestoreSensor type hints as mandatory
 (#163272)

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

diff --git a/pylint/plugins/hass_enforce_type_hints.py b/pylint/plugins/hass_enforce_type_hints.py
index 1c422c6cc45..da31c415828 100644
--- a/pylint/plugins/hass_enforce_type_hints.py
+++ b/pylint/plugins/hass_enforce_type_hints.py
@@ -781,14 +781,17 @@ _RESTORE_ENTITY_MATCH: list[TypeHintMatch] = [
     TypeHintMatch(
         function_name="async_get_last_state",
         return_type=["State", None],
+        mandatory=True,
     ),
     TypeHintMatch(
         function_name="async_get_last_extra_data",
         return_type=["ExtraStoredData", None],
+        mandatory=True,
     ),
     TypeHintMatch(
         function_name="extra_restore_state_data",
         return_type=["ExtraStoredData", None],
+        mandatory=True,
     ),
 ]
 _TOGGLE_ENTITY_MATCH: list[TypeHintMatch] = [
@@ -2559,10 +2562,12 @@ _INHERITANCE_MATCH: dict[str, list[ClassTypeHintMatch]] = {
                 TypeHintMatch(
                     function_name="extra_restore_state_data",
                     return_type="SensorExtraStoredData",
+                    mandatory=True,
                 ),
                 TypeHintMatch(
                     function_name="async_get_last_sensor_data",
                     return_type=["SensorExtraStoredData", None],
+                    mandatory=True,
                 ),
             ],
         ),