From d66056cfabd2d921208d2912254cb04f4d5c1b08 Mon Sep 17 00:00:00 2001 From: Marc Mueller <30130371+cdce8p@users.noreply.github.com> Date: Mon, 1 May 2023 21:11:40 +0200 Subject: [PATCH] Improve fritz generic typing (#92327) --- homeassistant/components/fritz/common.py | 3 +-- homeassistant/components/fritz/switch.py | 3 +-- 2 files changed, 2 insertions(+), 4 deletions(-) diff --git a/homeassistant/components/fritz/common.py b/homeassistant/components/fritz/common.py index 821b53f7e12..818030965f9 100644 --- a/homeassistant/components/fritz/common.py +++ b/homeassistant/components/fritz/common.py @@ -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 diff --git a/homeassistant/components/fritz/switch.py b/homeassistant/components/fritz/switch.py index c8a7952ae2b..5b8c4048530 100644 --- a/homeassistant/components/fritz/switch.py +++ b/homeassistant/components/fritz/switch.py @@ -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