diff --git a/homeassistant/components/alarm_control_panel/__init__.py b/homeassistant/components/alarm_control_panel/__init__.py index 4d74a39d977..8f523d5af7c 100644 --- a/homeassistant/components/alarm_control_panel/__init__.py +++ b/homeassistant/components/alarm_control_panel/__init__.py @@ -133,7 +133,7 @@ class AlarmControlPanelEntity(Entity): _attr_changed_by: str | None = None _attr_code_arm_required: bool = True _attr_code_format: CodeFormat | None = None - _attr_supported_features: int + _attr_supported_features: AlarmControlPanelEntityFeature | int @property def code_format(self) -> CodeFormat | None: @@ -207,7 +207,7 @@ class AlarmControlPanelEntity(Entity): await self.hass.async_add_executor_job(self.alarm_arm_custom_bypass, code) @property - def supported_features(self) -> int: + def supported_features(self) -> AlarmControlPanelEntityFeature | int: """Return the list of supported features.""" return self._attr_supported_features diff --git a/homeassistant/components/mqtt/alarm_control_panel.py b/homeassistant/components/mqtt/alarm_control_panel.py index 8a15c2587f1..136fe79752e 100644 --- a/homeassistant/components/mqtt/alarm_control_panel.py +++ b/homeassistant/components/mqtt/alarm_control_panel.py @@ -243,7 +243,7 @@ class MqttAlarm(MqttEntity, alarm.AlarmControlPanelEntity): return self._state @property - def supported_features(self) -> int: + def supported_features(self) -> AlarmControlPanelEntityFeature: """Return the list of supported features.""" return ( AlarmControlPanelEntityFeature.ARM_HOME diff --git a/homeassistant/components/overkiz/alarm_control_panel.py b/homeassistant/components/overkiz/alarm_control_panel.py index 2229c583297..ae7d16aee9c 100644 --- a/homeassistant/components/overkiz/alarm_control_panel.py +++ b/homeassistant/components/overkiz/alarm_control_panel.py @@ -39,7 +39,7 @@ from .entity import OverkizDescriptiveEntity class OverkizAlarmDescriptionMixin: """Define an entity description mixin for switch entities.""" - supported_features: int + supported_features: AlarmControlPanelEntityFeature fn_state: Callable[[Callable[[str], OverkizStateType]], str] diff --git a/pylint/plugins/hass_enforce_type_hints.py b/pylint/plugins/hass_enforce_type_hints.py index 2774fc54b02..ddf3a3a5155 100644 --- a/pylint/plugins/hass_enforce_type_hints.py +++ b/pylint/plugins/hass_enforce_type_hints.py @@ -661,7 +661,7 @@ _INHERITANCE_MATCH: dict[str, list[ClassTypeHintMatch]] = { ), TypeHintMatch( function_name="supported_features", - return_type="int", + return_type=["AlarmControlPanelEntityFeature", "int"], ), TypeHintMatch( function_name="alarm_disarm",