From 4cac20f8351a599bbd15dba67d637a32f2ba9fc6 Mon Sep 17 00:00:00 2001 From: Marc Mueller <30130371+cdce8p@users.noreply.github.com> Date: Wed, 25 Oct 2023 16:22:19 +0200 Subject: [PATCH] Fix google_tasks generic typing (#102778) --- homeassistant/components/google_tasks/coordinator.py | 2 +- homeassistant/components/google_tasks/todo.py | 6 ++++-- 2 files changed, 5 insertions(+), 3 deletions(-) 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