Use generic in iaqualink entity (#144989)

homeassistant/components/iaqualink/binary_sensor.py

@@ -34,7 +34,9 @@ async def async_setup_entry(
     )
 
 
-class HassAqualinkBinarySensor(AqualinkEntity, BinarySensorEntity):
+class HassAqualinkBinarySensor(
+    AqualinkEntity[AqualinkBinarySensor], BinarySensorEntity
+):
     """Representation of a binary sensor."""
 
     def __init__(self, dev: AqualinkBinarySensor) -> None:

homeassistant/components/iaqualink/climate.py

@@ -42,7 +42,7 @@ async def async_setup_entry(
     )
 
 
-class HassAqualinkThermostat(AqualinkEntity, ClimateEntity):
+class HassAqualinkThermostat(AqualinkEntity[AqualinkThermostat], ClimateEntity):
     """Representation of a thermostat."""
 
     _attr_hvac_modes = [HVACMode.HEAT, HVACMode.OFF]

homeassistant/components/iaqualink/entity.py

@@ -11,7 +11,7 @@ from homeassistant.helpers.entity import Entity
 from .const import DOMAIN
 
 
-class AqualinkEntity(Entity):
+class AqualinkEntity[AqualinkDeviceT: AqualinkDevice](Entity):
     """Abstract class for all Aqualink platforms.
 
     Entity state is updated via the interval timer within the integration.
@@ -23,7 +23,7 @@ class AqualinkEntity(Entity):
 
     _attr_should_poll = False
 
-    def __init__(self, dev: AqualinkDevice) -> None:
+    def __init__(self, dev: AqualinkDeviceT) -> None:
         """Initialize the entity."""
         self.dev = dev
         self._attr_unique_id = f"{dev.system.serial}_{dev.name}"

homeassistant/components/iaqualink/light.py

@@ -38,7 +38,7 @@ async def async_setup_entry(
     )
 
 
-class HassAqualinkLight(AqualinkEntity, LightEntity):
+class HassAqualinkLight(AqualinkEntity[AqualinkLight], LightEntity):
     """Representation of a light."""
 
     def __init__(self, dev: AqualinkLight) -> None:

homeassistant/components/iaqualink/sensor.py

@@ -32,7 +32,7 @@ async def async_setup_entry(
     )
 
 
-class HassAqualinkSensor(AqualinkEntity, SensorEntity):
+class HassAqualinkSensor(AqualinkEntity[AqualinkSensor], SensorEntity):
     """Representation of a sensor."""
 
     def __init__(self, dev: AqualinkSensor) -> None:

homeassistant/components/iaqualink/switch.py

@@ -31,7 +31,7 @@ async def async_setup_entry(
     )
 
 
-class HassAqualinkSwitch(AqualinkEntity, SwitchEntity):
+class HassAqualinkSwitch(AqualinkEntity[AqualinkSwitch], SwitchEntity):
     """Representation of a switch."""
 
     def __init__(self, dev: AqualinkSwitch) -> None: