From 9805d8a6cc43052e8dbeb4e2298487eaaa003369 Mon Sep 17 00:00:00 2001 From: Franck Nijhof Date: Tue, 29 Nov 2022 14:07:30 +0100 Subject: [PATCH] Fix device class typing in Universal Devices ISY994 (#82924) --- homeassistant/components/isy994/binary_sensor.py | 14 ++++++++------ 1 file changed, 8 insertions(+), 6 deletions(-) diff --git a/homeassistant/components/isy994/binary_sensor.py b/homeassistant/components/isy994/binary_sensor.py index 23e77ba849d..81efdb5922a 100644 --- a/homeassistant/components/isy994/binary_sensor.py +++ b/homeassistant/components/isy994/binary_sensor.py @@ -70,7 +70,7 @@ async def async_setup_entry( | ISYBinarySensorHeartbeat | ISYBinarySensorProgramEntity, ] = {} - child_nodes: list[tuple[Node, str | None, str | None]] = [] + child_nodes: list[tuple[Node, BinarySensorDeviceClass | None, str | None]] = [] entity: ISYInsteonBinarySensorEntity | ISYBinarySensorEntity | ISYBinarySensorHeartbeat | ISYBinarySensorProgramEntity hass_isy_data = hass.data[ISY994_DOMAIN][entry.entry_id] @@ -192,7 +192,9 @@ async def async_setup_entry( async_add_entities(entities) -def _detect_device_type_and_class(node: Group | Node) -> tuple[str | None, str | None]: +def _detect_device_type_and_class( + node: Group | Node, +) -> tuple[BinarySensorDeviceClass | None, str | None]: try: device_type = node.type except AttributeError: @@ -220,7 +222,7 @@ class ISYBinarySensorEntity(ISYNodeEntity, BinarySensorEntity): def __init__( self, node: Node, - force_device_class: str | None = None, + force_device_class: BinarySensorDeviceClass | None = None, unknown_state: bool | None = None, ) -> None: """Initialize the ISY994 binary sensor device.""" @@ -235,7 +237,7 @@ class ISYBinarySensorEntity(ISYNodeEntity, BinarySensorEntity): return bool(self._node.status) @property - def device_class(self) -> str | None: + def device_class(self) -> BinarySensorDeviceClass | None: """Return the class of this device. This was discovered by parsing the device type code during init @@ -255,7 +257,7 @@ class ISYInsteonBinarySensorEntity(ISYBinarySensorEntity): def __init__( self, node: Node, - force_device_class: str | None = None, + force_device_class: BinarySensorDeviceClass | None = None, unknown_state: bool | None = None, ) -> None: """Initialize the ISY994 binary sensor device.""" @@ -484,7 +486,7 @@ class ISYBinarySensorHeartbeat(ISYNodeEntity, BinarySensorEntity): return bool(self._computed_state) @property - def device_class(self) -> str: + def device_class(self) -> BinarySensorDeviceClass: """Get the class of this device.""" return BinarySensorDeviceClass.BATTERY