diff --git a/custom_components/sector/alarm_control_panel.py b/custom_components/sector/alarm_control_panel.py index ab0773d..378960c 100644 --- a/custom_components/sector/alarm_control_panel.py +++ b/custom_components/sector/alarm_control_panel.py @@ -42,7 +42,7 @@ async def async_setup_entry( async_add_entities([SectorAlarmControlPanel(coordinator)]) -class SectorAlarmControlPanel(CoordinatorEntity, AlarmControlPanelEntity): +class SectorAlarmControlPanel(CoordinatorEntity[SectorDataUpdateCoordinator], AlarmControlPanelEntity): """Representation of the Sector Alarm control panel.""" _attr_supported_features = ( diff --git a/custom_components/sector/binary_sensor.py b/custom_components/sector/binary_sensor.py index bc329fc..c1ec7a6 100644 --- a/custom_components/sector/binary_sensor.py +++ b/custom_components/sector/binary_sensor.py @@ -119,7 +119,7 @@ async def async_setup_entry( _LOGGER.debug("No binary sensor entities to add.") -class SectorAlarmBinarySensor(CoordinatorEntity, BinarySensorEntity): +class SectorAlarmBinarySensor(CoordinatorEntity[SectorDataUpdateCoordinator], BinarySensorEntity): """Representation of a Sector Alarm binary sensor.""" def __init__( @@ -185,7 +185,7 @@ def extra_state_attributes(self): } -class SectorAlarmPanelOnlineBinarySensor(CoordinatorEntity, BinarySensorEntity): +class SectorAlarmPanelOnlineBinarySensor(CoordinatorEntity[SectorDataUpdateCoordinator], BinarySensorEntity): """Representation of the Sector Alarm panel online status.""" def __init__( diff --git a/custom_components/sector/camera.py b/custom_components/sector/camera.py index 89a2d04..073047f 100644 --- a/custom_components/sector/camera.py +++ b/custom_components/sector/camera.py @@ -38,7 +38,7 @@ async def async_setup_entry( _LOGGER.debug("No camera entities to add.") -class SectorAlarmCamera(CoordinatorEntity, Camera): +class SectorAlarmCamera(CoordinatorEntity[SectorDataUpdateCoordinator], Camera): """Representation of a Sector Alarm camera.""" def __init__(self, coordinator: SectorDataUpdateCoordinator, camera_data: dict): diff --git a/custom_components/sector/lock.py b/custom_components/sector/lock.py index bf9261b..024e546 100644 --- a/custom_components/sector/lock.py +++ b/custom_components/sector/lock.py @@ -34,7 +34,7 @@ async def async_setup_entry( _LOGGER.debug("No lock entities to add.") -class SectorAlarmLock(CoordinatorEntity, LockEntity): +class SectorAlarmLock(CoordinatorEntity[SectorDataUpdateCoordinator], LockEntity): """Representation of a Sector Alarm lock.""" def __init__(self, coordinator: SectorDataUpdateCoordinator, device_info: dict): diff --git a/custom_components/sector/sensor.py b/custom_components/sector/sensor.py index e3bac0e..d2ef1ac 100644 --- a/custom_components/sector/sensor.py +++ b/custom_components/sector/sensor.py @@ -91,7 +91,7 @@ async def async_setup_entry( _LOGGER.debug("No sensor entities to add.") -class SectorAlarmSensor(CoordinatorEntity, SensorEntity): +class SectorAlarmSensor(CoordinatorEntity[SectorDataUpdateCoordinator], SensorEntity): """Representation of a Sector Alarm sensor.""" def __init__( diff --git a/custom_components/sector/switch.py b/custom_components/sector/switch.py index 70a3ad6..aaa51fb 100644 --- a/custom_components/sector/switch.py +++ b/custom_components/sector/switch.py @@ -40,7 +40,7 @@ async def async_setup_entry( _LOGGER.debug("No switch entities to add.") -class SectorAlarmSwitch(CoordinatorEntity, SwitchEntity): +class SectorAlarmSwitch(CoordinatorEntity[SectorDataUpdateCoordinator], SwitchEntity): """Representation of a Sector Alarm smart plug.""" _attr_device_class = SwitchDeviceClass.OUTLET