Enforce SirenEntityFeature (#82464)

This commit is contained in:
epenet 2022-11-22 07:16:29 +01:00 committed by GitHub
parent 0cb3ec142e
commit ff5f1aee53
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 3 additions and 3 deletions

View File

@ -162,7 +162,7 @@ class SirenEntity(ToggleEntity):
entity_description: SirenEntityDescription entity_description: SirenEntityDescription
_attr_available_tones: list[int | str] | dict[int, str] | None _attr_available_tones: list[int | str] | dict[int, str] | None
_attr_supported_features: SirenEntityFeature | int = 0 _attr_supported_features: SirenEntityFeature = SirenEntityFeature(0)
@final @final
@property @property
@ -190,6 +190,6 @@ class SirenEntity(ToggleEntity):
return None return None
@property @property
def supported_features(self) -> SirenEntityFeature | int: def supported_features(self) -> SirenEntityFeature:
"""Return the list of supported features.""" """Return the list of supported features."""
return self._attr_supported_features return self._attr_supported_features

View File

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