Home-Assistant
/
core
mirror of https://github.com/home-assistant/core.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Enforce WaterHeaterEntityFeature (#82467)

pull/82567/head
epenet 2022-11-22 07:27:09 +01:00 committed by GitHub
parent ab26ea06b9
commit 4134d722da
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
homeassistant/components/water_heater/__init__.py
View File

@ -167,7 +167,7 @@ class WaterHeaterEntity(Entity):
_attr_operation_list: list[str] | None = None
_attr_precision: float
_attr_state: None = None
_attr_supported_features: WaterHeaterEntityFeature | int = 0
_attr_supported_features: WaterHeaterEntityFeature = WaterHeaterEntityFeature(0)
_attr_target_temperature_high: float | None = None
_attr_target_temperature_low: float | None = None
_attr_target_temperature: float | None = None
@ -338,7 +338,7 @@ class WaterHeaterEntity(Entity):
)
@property
def supported_features(self) -> WaterHeaterEntityFeature | int:
def supported_features(self) -> WaterHeaterEntityFeature:
"""Return the list of supported features."""
return self._attr_supported_features

2
pylint/plugins/hass_enforce_type_hints.py
View File

@ -2539,7 +2539,7 @@ _INHERITANCE_MATCH: dict[str, list[ClassTypeHintMatch]] = {
),
TypeHintMatch(
function_name="supported_features",
return_type=["WaterHeaterEntityFeature", "int"],
return_type="WaterHeaterEntityFeature",
),
TypeHintMatch(
function_name="target_temperature",