diff --git a/homeassistant/components/google_tasks/coordinator.py b/homeassistant/components/google_tasks/coordinator.py index ab03cd52ec8..5377e2be567 100644 --- a/homeassistant/components/google_tasks/coordinator.py +++ b/homeassistant/components/google_tasks/coordinator.py @@ -16,7 +16,7 @@ UPDATE_INTERVAL: Final = datetime.timedelta(minutes=30) TIMEOUT = 10 -class TaskUpdateCoordinator(DataUpdateCoordinator): +class TaskUpdateCoordinator(DataUpdateCoordinator[list[dict[str, Any]]]): """Coordinator for fetching Google Tasks for a Task List form the API.""" def __init__( diff --git a/homeassistant/components/google_tasks/todo.py b/homeassistant/components/google_tasks/todo.py index 62220303932..5d2da33da71 100644 --- a/homeassistant/components/google_tasks/todo.py +++ b/homeassistant/components/google_tasks/todo.py @@ -58,7 +58,9 @@ async def async_setup_entry( ) -class GoogleTaskTodoListEntity(CoordinatorEntity, TodoListEntity): +class GoogleTaskTodoListEntity( + CoordinatorEntity[TaskUpdateCoordinator], TodoListEntity +): """A To-do List representation of the Shopping List.""" _attr_has_entity_name = True @@ -89,7 +91,7 @@ class GoogleTaskTodoListEntity(CoordinatorEntity, TodoListEntity): summary=item["title"], uid=item["id"], status=TODO_STATUS_MAP.get( - item.get("status"), TodoItemStatus.NEEDS_ACTION + item.get("status"), TodoItemStatus.NEEDS_ACTION # type: ignore[arg-type] ), ) for item in self.coordinator.data