From 383a820ae10e62a3dbd9de8e9ff6cbfa34d50d8d Mon Sep 17 00:00:00 2001 From: epenet <6771947+epenet@users.noreply.github.com> Date: Thu, 28 Oct 2021 14:29:37 +0200 Subject: [PATCH] Use DeviceInfo in switcher-kis (#58579) Co-authored-by: epenet <epenet@users.noreply.github.com> --- homeassistant/components/switcher_kis/switch.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/homeassistant/components/switcher_kis/switch.py b/homeassistant/components/switcher_kis/switch.py index 8b93e422e2a..620a742f4e3 100644 --- a/homeassistant/components/switcher_kis/switch.py +++ b/homeassistant/components/switcher_kis/switch.py @@ -23,6 +23,7 @@ from homeassistant.helpers import ( entity_platform, ) from homeassistant.helpers.dispatcher import async_dispatcher_connect +from homeassistant.helpers.entity import DeviceInfo from homeassistant.helpers.entity_platform import AddEntitiesCallback from homeassistant.helpers.update_coordinator import CoordinatorEntity @@ -92,11 +93,11 @@ class SwitcherBaseSwitchEntity(CoordinatorEntity, SwitchEntity): # Entity class attributes self._attr_name = coordinator.name self._attr_unique_id = f"{coordinator.device_id}-{coordinator.mac_address}" - self._attr_device_info = { - "connections": { + self._attr_device_info = DeviceInfo( + connections={ (device_registry.CONNECTION_NETWORK_MAC, coordinator.mac_address) } - } + ) @callback def _handle_coordinator_update(self) -> None: