Electric-Special/ha-core
1
0
Fork 0
mirror of https://github.com/Electric-Special/ha-core.git synced 2026-03-21 03:03:17 +01:00
Code Issues Packages Projects Releases Wiki Activity

Mark RestoreEntity/RestoreSensor type hints as mandatory (#163272)

Browse Source
This commit is contained in:
epenet
2026-02-17 15:34:16 +01:00
committed by GitHub
parent c205785f4f
commit ff2f0ac320
1 changed files with 5 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
pylint/plugins/hass_enforce_type_hints.py
View File

@@ -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,
),
],
),