From d065ddc5c19d6193494a4f3426a4a43090bdfc7d Mon Sep 17 00:00:00 2001 From: epenet <6771947+epenet@users.noreply.github.com> Date: Thu, 28 Oct 2021 18:23:41 +0200 Subject: [PATCH] Use DeviceInfo in twentemilieu (#58608) Co-authored-by: epenet --- .../components/twentemilieu/sensor.py | 18 ++++++------------ 1 file changed, 6 insertions(+), 12 deletions(-) diff --git a/homeassistant/components/twentemilieu/sensor.py b/homeassistant/components/twentemilieu/sensor.py index 89c750ec865..14568fa5d87 100644 --- a/homeassistant/components/twentemilieu/sensor.py +++ b/homeassistant/components/twentemilieu/sensor.py @@ -12,13 +12,7 @@ from twentemilieu import ( from homeassistant.components.sensor import SensorEntity from homeassistant.config_entries import ConfigEntry -from homeassistant.const import ( - ATTR_IDENTIFIERS, - ATTR_MANUFACTURER, - ATTR_NAME, - CONF_ID, - DEVICE_CLASS_DATE, -) +from homeassistant.const import CONF_ID, DEVICE_CLASS_DATE from homeassistant.core import HomeAssistant, callback from homeassistant.exceptions import PlatformNotReady from homeassistant.helpers.dispatcher import async_dispatcher_connect @@ -153,8 +147,8 @@ class TwenteMilieuSensor(SensorEntity): @property def device_info(self) -> DeviceInfo: """Return device information about Twente Milieu.""" - return { - ATTR_IDENTIFIERS: {(DOMAIN, self._unique_id)}, - ATTR_NAME: "Twente Milieu", - ATTR_MANUFACTURER: "Twente Milieu", - } + return DeviceInfo( + identifiers={(DOMAIN, self._unique_id)}, + manufacturer="Twente Milieu", + name="Twente Milieu", + )