diff --git a/stdlib/2/itertools.pyi b/stdlib/2/itertools.pyi index 286d20055..da69d3dfe 100644 --- a/stdlib/2/itertools.pyi +++ b/stdlib/2/itertools.pyi @@ -49,11 +49,11 @@ _T3 = TypeVar('_T3') _T4 = TypeVar('_T4') @overload -def imap(func: Callable[[_T1], _S], iter1: Iterable[_T1]) -> Iterable[_S]: ... +def imap(func: Callable[[_T1], _S], iter1: Iterable[_T1]) -> Iterator[_S]: ... @overload def imap(func: Callable[[_T1, _T2], _S], iter1: Iterable[_T1], - iter2: Iterable[_T2]) -> Iterable[_S]: ... # TODO more than two iterables + iter2: Iterable[_T2]) -> Iterator[_S]: ... # TODO more than two iterables def starmap(func: Any, iterable: Iterable[Any]) -> Iterator[Any]: ... def takewhile(predicate: Callable[[_T], Any], @@ -61,16 +61,16 @@ def takewhile(predicate: Callable[[_T], Any], def tee(iterable: Iterable[Any], n: int = ...) -> Iterator[Any]: ... @overload -def izip(iter1: Iterable[_T1]) -> Iterable[Tuple[_T1]]: ... +def izip(iter1: Iterable[_T1]) -> Iterator[Tuple[_T1]]: ... @overload def izip(iter1: Iterable[_T1], - iter2: Iterable[_T2]) -> Iterable[Tuple[_T1, _T2]]: ... + iter2: Iterable[_T2]) -> Iterator[Tuple[_T1, _T2]]: ... @overload def izip(iter1: Iterable[_T1], iter2: Iterable[_T2], - iter3: Iterable[_T3]) -> Iterable[Tuple[_T1, _T2, _T3]]: ... + iter3: Iterable[_T3]) -> Iterator[Tuple[_T1, _T2, _T3]]: ... @overload def izip(iter1: Iterable[_T1], iter2: Iterable[_T2], iter3: Iterable[_T3], - iter4: Iterable[_T4]) -> Iterable[Tuple[_T1, _T2, + iter4: Iterable[_T4]) -> Iterator[Tuple[_T1, _T2, _T3, _T4]]: ... # TODO more than four iterables def izip_longest(*p: Iterable[Any], fillvalue: Any = ...) -> Iterator[Any]: ... @@ -82,6 +82,6 @@ def product(*p: Iterable[_T], repeat: int = ...) -> Iterator[Sequence[_T]]: ... def permutations(iterable: Iterable[_T], r: int = ...) -> Iterator[Sequence[_T]]: ... def combinations(iterable: Iterable[_T], - r: int) -> Iterable[Sequence[_T]]: ... + r: int) -> Iterator[Sequence[_T]]: ... def combinations_with_replacement(iterable: Iterable[_T], - r: int) -> Iterable[Sequence[_T]]: ... + r: int) -> Iterator[Sequence[_T]]: ...