diff --git a/pylint/plugins/hass_enforce_type_hints.py b/pylint/plugins/hass_enforce_type_hints.py index 1cd47c78899..b8dfe4ef3f5 100644 --- a/pylint/plugins/hass_enforce_type_hints.py +++ b/pylint/plugins/hass_enforce_type_hints.py @@ -698,6 +698,7 @@ _ENTITY_MATCH: list[TypeHintMatch] = [ TypeHintMatch( function_name="device_info", return_type=["DeviceInfo", None], + mandatory=True, ), TypeHintMatch( function_name="device_class",