diff --git a/stdlib/2.7/contextlib.pyi b/stdlib/2.7/contextlib.pyi index f9aa2945c..af9f0ab2a 100644 --- a/stdlib/2.7/contextlib.pyi +++ b/stdlib/2.7/contextlib.pyi @@ -2,16 +2,13 @@ # NOTE: These are incomplete! -from typing import Any, TypeVar, Callable, Generic, Iterator +from typing import Any, TypeVar, Generic + +# TODO more precise type? +def contextmanager(func: Any) -> Any: ... _T = TypeVar('_T') -class _GeneratorContextManager(Generic[_T]): - def __enter__(self) -> _T: ... - def __exit__(self, *exc_info) -> None: ... - -def contextmanager(func: Callable[..., Iterator[_T]]) -> Callable[..., _GeneratorContextManager[_T]]: ... - class closing(Generic[_T]): def __init__(self, thing: _T) -> None: ... def __enter__(self) -> _T: ...