From f43c802a03640b893472de5a376a1bcc3723044c Mon Sep 17 00:00:00 2001 From: Jeef Date: Wed, 12 Oct 2022 04:05:22 -0600 Subject: [PATCH] Flume code quality improvments (#79815) --- homeassistant/components/flume/binary_sensor.py | 10 ++++++---- homeassistant/components/flume/entity.py | 2 +- 2 files changed, 7 insertions(+), 5 deletions(-) diff --git a/homeassistant/components/flume/binary_sensor.py b/homeassistant/components/flume/binary_sensor.py index 235d7c3edd6..f6b1f63a26c 100644 --- a/homeassistant/components/flume/binary_sensor.py +++ b/homeassistant/components/flume/binary_sensor.py @@ -33,6 +33,11 @@ from .coordinator import ( from .entity import FlumeEntity from .util import get_valid_flume_devices +BINARY_SENSOR_DESCRIPTION_CONNECTED = BinarySensorEntityDescription( + name="Connected", + key="connected", +) + @dataclass class FlumeBinarySensorRequiredKeysMixin: @@ -93,10 +98,7 @@ async def async_setup_entry( connection_sensor = FlumeConnectionBinarySensor( coordinator=connection_coordinator, - description=BinarySensorEntityDescription( - name="Connected", - key="connected", - ), + description=BINARY_SENSOR_DESCRIPTION_CONNECTED, device_id=device_id, location_name=device_location_name, is_bridge=(device[KEY_DEVICE_TYPE] is FLUME_TYPE_BRIDGE), diff --git a/homeassistant/components/flume/entity.py b/homeassistant/components/flume/entity.py index 4aeba6d2bc6..7cd84127c64 100644 --- a/homeassistant/components/flume/entity.py +++ b/homeassistant/components/flume/entity.py @@ -10,7 +10,7 @@ from homeassistant.helpers.update_coordinator import ( from .const import DOMAIN -class FlumeEntity(CoordinatorEntity[DataUpdateCoordinator]): +class FlumeEntity(CoordinatorEntity[DataUpdateCoordinator[None]]): """Base entity class.""" _attr_attribution = "Data provided by Flume API"