From f9ed216ddf0bf1f1b9cf4f9e5763487ce746c185 Mon Sep 17 00:00:00 2001 From: epenet Date: Wed, 27 Oct 2021 12:09:42 +0000 Subject: [PATCH] Use DeviceInfo in soma --- homeassistant/components/soma/__init__.py | 14 +++++++------- 1 file changed, 7 insertions(+), 7 deletions(-) diff --git a/homeassistant/components/soma/__init__.py b/homeassistant/components/soma/__init__.py index 77ecd3af96bfe..532e6204ad90c 100644 --- a/homeassistant/components/soma/__init__.py +++ b/homeassistant/components/soma/__init__.py @@ -8,7 +8,7 @@ from homeassistant.const import CONF_HOST, CONF_PORT from homeassistant.core import HomeAssistant import homeassistant.helpers.config_validation as cv -from homeassistant.helpers.entity import Entity +from homeassistant.helpers.entity import DeviceInfo, Entity from .const import API, DOMAIN, HOST, PORT @@ -89,13 +89,13 @@ def name(self): return self.device["name"] @property - def device_info(self): + def device_info(self) -> DeviceInfo: """Return device specific attributes. Implemented by platform classes. """ - return { - "identifiers": {(DOMAIN, self.unique_id)}, - "name": self.name, - "manufacturer": "Wazombi Labs", - } + return DeviceInfo( + identifiers={(DOMAIN, self.unique_id)}, + manufacturer="Wazombi Labs", + name=self.name, + )