Improve fritz generic typing (#92327)

This commit is contained in:
Marc Mueller 2023-05-01 21:11:40 +02:00 committed by GitHub
parent 324df197d1
commit d66056cfab
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 2 additions and 4 deletions

View file

@ -1049,10 +1049,9 @@ class FritzEntityDescription(EntityDescription, FritzRequireKeysMixin):
"""Fritz entity base description."""
class FritzBoxBaseCoordinatorEntity(update_coordinator.CoordinatorEntity):
class FritzBoxBaseCoordinatorEntity(update_coordinator.CoordinatorEntity[AvmWrapper]):
"""Fritz host coordinator entity base class."""
coordinator: AvmWrapper
entity_description: FritzEntityDescription
_attr_has_entity_name = True

View file

@ -256,10 +256,9 @@ async def async_setup_entry(
)
class FritzBoxBaseCoordinatorSwitch(CoordinatorEntity, SwitchEntity):
class FritzBoxBaseCoordinatorSwitch(CoordinatorEntity[AvmWrapper], SwitchEntity):
"""Fritz switch coordinator base class."""
coordinator: AvmWrapper
entity_description: SwitchEntityDescription
_attr_has_entity_name = True