From d214bfec479836d3c9c2fc83c1e4ec898f348570 Mon Sep 17 00:00:00 2001 From: epenet <6771947+epenet@users.noreply.github.com> Date: Thu, 28 Oct 2021 18:26:53 +0200 Subject: [PATCH] Use DeviceInfo in upb (#58621) Co-authored-by: epenet --- homeassistant/components/upb/__init__.py | 18 +++++++++--------- 1 file changed, 9 insertions(+), 9 deletions(-) diff --git a/homeassistant/components/upb/__init__.py b/homeassistant/components/upb/__init__.py index a3c3016dc05..90d7c35cf64 100644 --- a/homeassistant/components/upb/__init__.py +++ b/homeassistant/components/upb/__init__.py @@ -4,7 +4,7 @@ import upb_lib from homeassistant.const import ATTR_COMMAND, CONF_FILE_PATH, CONF_HOST from homeassistant.core import callback -from homeassistant.helpers.entity import Entity +from homeassistant.helpers.entity import DeviceInfo, Entity from .const import ( ATTR_ADDRESS, @@ -119,12 +119,12 @@ class UpbAttachedEntity(UpbEntity): """Base class for UPB attached entities.""" @property - def device_info(self): + def device_info(self) -> DeviceInfo: """Device info for the entity.""" - return { - "name": self._element.name, - "identifiers": {(DOMAIN, self._element.index)}, - "sw_version": self._element.version, - "manufacturer": self._element.manufacturer, - "model": self._element.product, - } + return DeviceInfo( + identifiers={(DOMAIN, self._element.index)}, + manufacturer=self._element.manufacturer, + model=self._element.product, + name=self._element.name, + sw_version=self._element.version, + )