Adjust type hints for RemoteEntityFeature (#82265)

This commit is contained in:
epenet 2022-11-17 15:06:16 +01:00 committed by GitHub
parent b6586d5c34
commit 9b9bdc725f
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 6 additions and 6 deletions

View file

@ -166,10 +166,10 @@ class RemoteEntity(ToggleEntity):
entity_description: RemoteEntityDescription
_attr_activity_list: list[str] | None = None
_attr_current_activity: str | None = None
_attr_supported_features: int = 0
_attr_supported_features: RemoteEntityFeature | int = 0
@property
def supported_features(self) -> int:
def supported_features(self) -> RemoteEntityFeature | int:
"""Flag supported features."""
return self._attr_supported_features

View file

@ -2052,16 +2052,16 @@ _INHERITANCE_MATCH: dict[str, list[ClassTypeHintMatch]] = {
base_class="RemoteEntity",
matches=[
TypeHintMatch(
function_name="supported_features",
return_type="int",
function_name="activity_list",
return_type=["list[str]", None],
),
TypeHintMatch(
function_name="current_activity",
return_type=["str", None],
),
TypeHintMatch(
function_name="activity_list",
return_type=["list[str]", None],
function_name="supported_features",
return_type=["RemoteEntityFeature", "int"],
),
TypeHintMatch(
function_name="send_command",