Fix support old Hass versions

AlexxIT · Apr 16, 2024 · 1351306 · 1351306
1 parent 7488e1a
commit 1351306
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/custom_components/yandex_station/camera.py b/custom_components/yandex_station/camera.py
@@ -8,7 +8,7 @@
 from homeassistant.components.media_player import MediaPlayerState, MediaType
 from homeassistant.const import CONTENT_TYPE_MULTIPART
 from homeassistant.helpers.aiohttp_client import async_get_clientsession
-from homeassistant.helpers.device_registry import DeviceInfo
+from homeassistant.helpers.entity import DeviceInfo
 
 from .core.const import DOMAIN
 from .core.image import draw_cover, draw_lyrics, draw_none

diff --git a/custom_components/yandex_station/core/entity.py b/custom_components/yandex_station/core/entity.py
@@ -1,7 +1,6 @@
 import logging
 
-from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.entity import Entity
+from homeassistant.helpers.entity import DeviceInfo, Entity
 
 from .const import DOMAIN
 from .yandex_quasar import YandexQuasar