mirror of https://github.com/home-assistant/core
Enable inheritance checks on ExtraStoredData (#90021)
This commit is contained in:
parent
43ce6f843c
commit
292feb4e24
|
@ -132,7 +132,7 @@ class ZWaveNodeFirmwareUpdate(UpdateEntity):
|
|||
self._attr_device_info = get_device_info(driver, node)
|
||||
|
||||
@property
|
||||
def extra_restore_state_data(self) -> ExtraStoredData:
|
||||
def extra_restore_state_data(self) -> ZWaveNodeFirmwareUpdateExtraStoredData:
|
||||
"""Return ZWave Node Firmware Update specific state data to be restored."""
|
||||
return ZWaveNodeFirmwareUpdateExtraStoredData(self._latest_version_firmware)
|
||||
|
||||
|
|
|
@ -680,6 +680,7 @@ _RESTORE_ENTITY_MATCH: list[TypeHintMatch] = [
|
|||
TypeHintMatch(
|
||||
function_name="extra_restore_state_data",
|
||||
return_type=["ExtraStoredData", None],
|
||||
check_return_type_inheritance=True,
|
||||
),
|
||||
]
|
||||
_TOGGLE_ENTITY_MATCH: list[TypeHintMatch] = [
|
||||
|
|
Loading…
Reference in New Issue