diff --git a/stdlib/3.4/asyncio/__init__.pyi b/stdlib/3.4/asyncio/__init__.pyi index a25b6eec2..60a7ddeae 100644 --- a/stdlib/3.4/asyncio/__init__.pyi +++ b/stdlib/3.4/asyncio/__init__.pyi @@ -41,6 +41,7 @@ from asyncio.futures import ( CancelledError as CancelledError, TimeoutError as TimeoutError, InvalidStateError as InvalidStateError, + wrap_future as wrap_future, ) from asyncio.tasks import ( FIRST_COMPLETED as FIRST_COMPLETED, diff --git a/stdlib/3.4/asyncio/futures.pyi b/stdlib/3.4/asyncio/futures.pyi index b88009324..f651c66b1 100644 --- a/stdlib/3.4/asyncio/futures.pyi +++ b/stdlib/3.4/asyncio/futures.pyi @@ -6,6 +6,7 @@ from concurrent.futures._base import ( from concurrent.futures import ( CancelledError as CancelledError, TimeoutError as TimeoutError, + Future as ConcurrentFuture, ) __all__ = ... # type: str @@ -44,3 +45,5 @@ class Future(Iterable[_T], Awaitable[_T], Generic[_T]): def _copy_state(self, other: Any) -> None: ... def __iter__(self) -> Generator[Any, None, _T]: ... def __await__(self) -> Generator[Any, None, _T]: ... + +def wrap_future(f: Union[ConcurrentFuture[_T], Future[_T]]) -> Future[_T]: ...