From 15a0c28530ffcda0d4be2026511af3604336f38f Mon Sep 17 00:00:00 2001 From: Nikita Sobolev Date: Sun, 14 May 2023 15:32:27 +0300 Subject: [PATCH] Use `_CoroutineLike` in `asyncio.taskgroups` (#10180) --- stdlib/asyncio/taskgroups.pyi | 8 +++----- 1 file changed, 3 insertions(+), 5 deletions(-) diff --git a/stdlib/asyncio/taskgroups.pyi b/stdlib/asyncio/taskgroups.pyi index 8daa96f1e..08ea8f665 100644 --- a/stdlib/asyncio/taskgroups.pyi +++ b/stdlib/asyncio/taskgroups.pyi @@ -1,11 +1,11 @@ # This only exists in 3.11+. See VERSIONS. -from collections.abc import Coroutine, Generator from contextvars import Context from types import TracebackType -from typing import Any, TypeVar +from typing import TypeVar from typing_extensions import Self +from . import _CoroutineLike from .tasks import Task __all__ = ["TaskGroup"] @@ -15,6 +15,4 @@ _T = TypeVar("_T") class TaskGroup: async def __aenter__(self) -> Self: ... async def __aexit__(self, et: type[BaseException] | None, exc: BaseException | None, tb: TracebackType | None) -> None: ... - def create_task( - self, coro: Generator[Any, None, _T] | Coroutine[Any, Any, _T], *, name: str | None = None, context: Context | None = None - ) -> Task[_T]: ... + def create_task(self, coro: _CoroutineLike[_T], *, name: str | None = None, context: Context | None = None) -> Task[_T]: ...