diff --git a/homeassistant/components/discovergy/sensor.py b/homeassistant/components/discovergy/sensor.py index 35955a6b189..3f4069752f2 100644 --- a/homeassistant/components/discovergy/sensor.py +++ b/homeassistant/components/discovergy/sensor.py @@ -11,16 +11,13 @@ from homeassistant.components.sensor import ( ) from homeassistant.config_entries import ConfigEntry from homeassistant.const import ( - ATTR_IDENTIFIERS, - ATTR_MANUFACTURER, - ATTR_MODEL, - ATTR_NAME, UnitOfElectricPotential, UnitOfEnergy, UnitOfPower, UnitOfVolume, ) from homeassistant.core import HomeAssistant +from homeassistant.helpers.entity import DeviceInfo from homeassistant.helpers.entity_platform import AddEntitiesCallback from homeassistant.helpers.typing import StateType from homeassistant.helpers.update_coordinator import CoordinatorEntity @@ -198,12 +195,12 @@ class DiscovergySensor(CoordinatorEntity[DiscovergyUpdateCoordinator], SensorEnt self.entity_description = description self._attr_unique_id = f"{meter.full_serial_number}-{data_key}" - self._attr_device_info = { - ATTR_IDENTIFIERS: {(DOMAIN, meter.get_meter_id())}, - ATTR_NAME: f"{meter.measurement_type.capitalize()} {meter.location.street} {meter.location.street_number}", - ATTR_MODEL: f"{meter.type} {meter.full_serial_number}", - ATTR_MANUFACTURER: MANUFACTURER, - } + self._attr_device_info = DeviceInfo( + identifiers={(DOMAIN, meter.get_meter_id())}, + name=f"{meter.measurement_type.capitalize()} {meter.location.street} {meter.location.street_number}", + model=f"{meter.type} {meter.full_serial_number}", + manufacturer=MANUFACTURER, + ) @property def native_value(self) -> StateType: