evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Improve flume generic typing (#85329)

Browse source
This commit is contained in:
Marc Mueller 2023-01-07 14:10:44 +01:00 • committed by GitHub
parent dddba4ba45
commit b27e89b40e
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
3 changed files with 26 additions and 13 deletions
Download patch file Download diff file Expand all files Collapse all files

10
homeassistant/components/flume/binary_sensor.py
View file

@ -125,11 +125,12 @@ async def async_setup_entry(
async_add_entities(flume_entity_list)
class FlumeNotificationBinarySensor(FlumeEntity, BinarySensorEntity):
class FlumeNotificationBinarySensor(
FlumeEntity[FlumeNotificationDataUpdateCoordinator], BinarySensorEntity
):
"""Binary sensor class."""
entity_description: FlumeBinarySensorEntityDescription
coordinator: FlumeNotificationDataUpdateCoordinator
@property
def is_on(self) -> bool:
@ -144,11 +145,12 @@ class FlumeNotificationBinarySensor(FlumeEntity, BinarySensorEntity):
)
class FlumeConnectionBinarySensor(FlumeEntity, BinarySensorEntity):
class FlumeConnectionBinarySensor(
FlumeEntity[FlumeDeviceConnectionUpdateCoordinator], BinarySensorEntity
):
"""Binary Sensor class for WIFI Connection status."""
entity_description: FlumeBinarySensorEntityDescription
coordinator: FlumeDeviceConnectionUpdateCoordinator
_attr_entity_category = EntityCategory.DIAGNOSTIC
_attr_device_class = BinarySensorDeviceClass.CONNECTIVITY