Fix ConfigEntry stragglers

bang-olufsen · Oct 28, 2024 · 41332cc · 41332cc
1 parent ae1b8f3
commit 41332cc
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 6 deletions.
diff --git 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
@@ -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