diff --git a/homeassistant/components/zwave_js/entity.py b/homeassistant/components/zwave_js/entity.py index aa915cd5822..cf15f32932b 100644 --- a/homeassistant/components/zwave_js/entity.py +++ b/homeassistant/components/zwave_js/entity.py @@ -10,7 +10,7 @@ from zwave_js_server.model.value import Value as ZwaveValue, get_value_id from homeassistant.config_entries import ConfigEntry from homeassistant.core import callback from homeassistant.helpers.dispatcher import async_dispatcher_connect -from homeassistant.helpers.entity import Entity +from homeassistant.helpers.entity import DeviceInfo, Entity from .const import DOMAIN from .discovery import ZwaveDiscoveryInfo @@ -54,9 +54,9 @@ class ZWaveBaseEntity(Entity): ) self._attr_assumed_state = self.info.assumed_state # device is precreated in main handler - self._attr_device_info = { - "identifiers": {get_device_id(self.client, self.info.node)}, - } + self._attr_device_info = DeviceInfo( + identifiers={get_device_id(self.client, self.info.node)}, + ) @callback def on_value_update(self) -> None: diff --git a/homeassistant/components/zwave_js/sensor.py b/homeassistant/components/zwave_js/sensor.py index 715affe351e..a3d06a21f89 100644 --- a/homeassistant/components/zwave_js/sensor.py +++ b/homeassistant/components/zwave_js/sensor.py @@ -47,6 +47,7 @@ from homeassistant.const import ( from homeassistant.core import HomeAssistant, callback from homeassistant.helpers import entity_platform from homeassistant.helpers.dispatcher import async_dispatcher_connect +from homeassistant.helpers.entity import DeviceInfo from homeassistant.helpers.entity_platform import AddEntitiesCallback from .const import ( @@ -476,9 +477,9 @@ class ZWaveNodeStatusSensor(SensorEntity): f"{self.client.driver.controller.home_id}.{node.node_id}.node_status" ) # device is precreated in main handler - self._attr_device_info = { - "identifiers": {get_device_id(self.client, self.node)}, - } + self._attr_device_info = DeviceInfo( + identifiers={get_device_id(self.client, self.node)}, + ) self._attr_native_value: str = node.status.name.lower() async def async_poll_value(self, _: bool) -> None: