diff --git a/homeassistant/components/demo/humidifier.py b/homeassistant/components/demo/humidifier.py index 9b55583d12b..772726ac1d5 100644 --- a/homeassistant/components/demo/humidifier.py +++ b/homeassistant/components/demo/humidifier.py @@ -13,7 +13,7 @@ from homeassistant.core import HomeAssistant from homeassistant.helpers.entity_platform import AddEntitiesCallback from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType -SUPPORT_FLAGS = 0 +SUPPORT_FLAGS = HumidifierEntityFeature(0) async def async_setup_platform( diff --git a/homeassistant/components/humidifier/__init__.py b/homeassistant/components/humidifier/__init__.py index 976dad03a45..61d7b6c3944 100644 --- a/homeassistant/components/humidifier/__init__.py +++ b/homeassistant/components/humidifier/__init__.py @@ -137,7 +137,7 @@ class HumidifierEntity(ToggleEntity): _attr_max_humidity: int = DEFAULT_MAX_HUMIDITY _attr_min_humidity: int = DEFAULT_MIN_HUMIDITY _attr_mode: str | None - _attr_supported_features: HumidifierEntityFeature | int = 0 + _attr_supported_features: HumidifierEntityFeature = HumidifierEntityFeature(0) _attr_target_humidity: int | None = None @property @@ -224,6 +224,6 @@ class HumidifierEntity(ToggleEntity): return self._attr_max_humidity @property - def supported_features(self) -> HumidifierEntityFeature | int: + def supported_features(self) -> HumidifierEntityFeature: """Return the list of supported features.""" return self._attr_supported_features diff --git a/pylint/plugins/hass_enforce_type_hints.py b/pylint/plugins/hass_enforce_type_hints.py index 0ae157a694f..4c15de25c4d 100644 --- a/pylint/plugins/hass_enforce_type_hints.py +++ b/pylint/plugins/hass_enforce_type_hints.py @@ -1429,7 +1429,7 @@ _INHERITANCE_MATCH: dict[str, list[ClassTypeHintMatch]] = { ), TypeHintMatch( function_name="supported_features", - return_type=["HumidifierEntityFeature", "int"], + return_type="HumidifierEntityFeature", ), TypeHintMatch( function_name="target_humidity",