From ab07bc5298fcd0b8458db176997e947fa0f0029e Mon Sep 17 00:00:00 2001 From: Marc Mueller <30130371+cdce8p@users.noreply.github.com> Date: Thu, 16 May 2024 12:47:43 +0200 Subject: [PATCH] Improve ReloadServiceHelper typing (#117552) --- homeassistant/helpers/service.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/homeassistant/helpers/service.py b/homeassistant/helpers/service.py index 1f3d59e761c..bc6bef3f0ed 100644 --- a/homeassistant/helpers/service.py +++ b/homeassistant/helpers/service.py @@ -3,13 +3,13 @@ from __future__ import annotations import asyncio -from collections.abc import Awaitable, Callable, Iterable +from collections.abc import Awaitable, Callable, Coroutine, Iterable import dataclasses from enum import Enum from functools import cache, partial import logging from types import ModuleType -from typing import TYPE_CHECKING, Any, TypedDict, TypeGuard, TypeVar, cast +from typing import TYPE_CHECKING, Any, Generic, TypedDict, TypeGuard, TypeVar, cast import voluptuous as vol @@ -1156,7 +1156,7 @@ def verify_domain_control( return decorator -class ReloadServiceHelper: +class ReloadServiceHelper(Generic[_T]): """Helper for reload services. The helper has the following purposes: @@ -1166,7 +1166,7 @@ class ReloadServiceHelper: def __init__( self, - service_func: Callable[[ServiceCall], Awaitable], + service_func: Callable[[ServiceCall], Coroutine[Any, Any, Any]], reload_targets_func: Callable[[ServiceCall], set[_T]], ) -> None: """Initialize ReloadServiceHelper."""