From 95f7b0c026d5045fa881cee4cd3f1cf6fd036b32 Mon Sep 17 00:00:00 2001
From: epenet <6771947+epenet@users.noreply.github.com>
Date: Thu, 28 Oct 2021 23:58:53 +0200
Subject: [PATCH] Use DeviceInfo in waze-travel-time (#58637)
---
homeassistant/components/waze_travel_time/sensor.py | 11 ++++++-----
1 file changed, 6 insertions(+), 5 deletions(-)
diff --git a/homeassistant/components/waze_travel_time/sensor.py b/homeassistant/components/waze_travel_time/sensor.py
index 81ee48ebd2f..8612f3ee54b 100644
--- a/homeassistant/components/waze_travel_time/sensor.py
+++ b/homeassistant/components/waze_travel_time/sensor.py
@@ -22,6 +22,7 @@ from homeassistant.const import (
)
from homeassistant.core import Config, CoreState, HomeAssistant
import homeassistant.helpers.config_validation as cv
+from homeassistant.helpers.entity import DeviceInfo
from homeassistant.helpers.entity_platform import AddEntitiesCallback
from homeassistant.helpers.typing import DiscoveryInfoType
@@ -163,11 +164,11 @@ class WazeTravelTime(SensorEntity):
"""Representation of a Waze travel time sensor."""
_attr_native_unit_of_measurement = TIME_MINUTES
- _attr_device_info = {
- "name": "Waze",
- "identifiers": {(DOMAIN, DOMAIN)},
- "entry_type": "service",
- }
+ _attr_device_info = DeviceInfo(
+ entry_type="service",
+ name="Waze",
+ identifiers={(DOMAIN, DOMAIN)},
+ )
def __init__(self, unique_id, name, origin, destination, waze_data):
"""Initialize the Waze travel time sensor."""