From 41332cc4432d2ff642ba6b55f284010728eadd79 Mon Sep 17 00:00:00 2001 From: mj23000 Date: Mon, 28 Oct 2024 14:52:47 +0100 Subject: [PATCH] Fix ConfigEntry stragglers --- custom_components/bang_olufsen/event.py | 7 +++---- custom_components/bang_olufsen/media_player.py | 3 +-- 2 files changed, 4 insertions(+), 6 deletions(-) diff --git a/custom_components/bang_olufsen/event.py b/custom_components/bang_olufsen/event.py index 50719c1..bad977a 100644 --- a/custom_components/bang_olufsen/event.py +++ b/custom_components/bang_olufsen/event.py @@ -5,7 +5,6 @@ from mozart_api.models import PairedRemote from homeassistant.components.event import EventDeviceClass, EventEntity -from homeassistant.config_entries import ConfigEntry from homeassistant.const import CONF_MODEL from homeassistant.core import HomeAssistant, callback from homeassistant.helpers.device_registry import DeviceInfo @@ -105,7 +104,7 @@ class BangOlufsenButtonEvent(BangOlufsenEvent): _attr_event_types = DEVICE_BUTTON_EVENTS _attr_icon = "mdi:gesture-tap-button" - def __init__(self, config_entry: ConfigEntry, button_type: str) -> None: + def __init__(self, config_entry: BangOlufsenConfigEntry, button_type: str) -> None: """Initialize Button.""" super().__init__(config_entry) @@ -143,7 +142,7 @@ class BangOlufsenRemoteKeyEvent(BangOlufsenEvent): def __init__( self, - config_entry: ConfigEntry, + config_entry: BangOlufsenConfigEntry, remote: PairedRemote, key_type: str, ) -> None: @@ -187,7 +186,7 @@ class BangOlufsenEventProximity(BangOlufsenEvent): _attr_icon = "mdi:account-question" _attr_translation_key = "proximity" - def __init__(self, config_entry: ConfigEntry) -> None: + def __init__(self, config_entry: BangOlufsenConfigEntry) -> None: """Init the proximity event.""" super().__init__(config_entry) diff --git a/custom_components/bang_olufsen/media_player.py b/custom_components/bang_olufsen/media_player.py index 3775899..0f69262 100644 --- a/custom_components/bang_olufsen/media_player.py +++ b/custom_components/bang_olufsen/media_player.py @@ -53,7 +53,6 @@ RepeatMode, async_process_play_media_url, ) -from homeassistant.config_entries import ConfigEntry from homeassistant.const import CONF_MODEL, Platform from homeassistant.core import ( HomeAssistant, @@ -233,7 +232,7 @@ class BangOlufsenMediaPlayer(MediaPlayerEntity, BangOlufsenEntity): _attr_icon = "mdi:speaker-wireless" _attr_name: None | str = None - def __init__(self, config_entry: ConfigEntry) -> None: + def __init__(self, config_entry: BangOlufsenConfigEntry) -> None: """Initialize the media player.""" super().__init__(config_entry) self._attr_should_poll = True