diff --git a/stdlib/2/heapq.pyi b/stdlib/2/heapq.pyi index 4a7a65f62..00abb31a6 100644 --- a/stdlib/2/heapq.pyi +++ b/stdlib/2/heapq.pyi @@ -1,4 +1,4 @@ -from typing import TypeVar, List, Iterable, Any, Callable +from typing import TypeVar, List, Iterable, Any, Callable, Optional _T = TypeVar('_T') @@ -11,5 +11,6 @@ def heapify(x: List[_T]) -> None: ... def heapreplace(heap: List[_T], item: _T) -> _T: raise IndexError() # if heap is empty def merge(*iterables: Iterable[_T]) -> Iterable[_T]: ... -def nlargest(n: int, iterable: Iterable[_T]) -> List[_T]: ... +def nlargest(n: int, iterable: Iterable[_T], + key: Optional[Callable[[_T], Any]] = ...) -> List[_T]: ... def nsmallest(n: int, iterable: Iterable[_T]) -> List[_T]: ... diff --git a/stdlib/3/heapq.pyi b/stdlib/3/heapq.pyi index f56ae7204..5c49dfac1 100644 --- a/stdlib/3/heapq.pyi +++ b/stdlib/3/heapq.pyi @@ -3,7 +3,7 @@ # Based on http://docs.python.org/3.2/library/heapq.html import sys -from typing import TypeVar, List, Iterable, Any, Callable +from typing import TypeVar, List, Iterable, Any, Callable, Optional _T = TypeVar('_T') @@ -18,6 +18,6 @@ if sys.version_info >= (3, 5): else: def merge(*iterables: Iterable[_T]) -> Iterable[_T]: ... def nlargest(n: int, iterable: Iterable[_T], - key: Callable[[_T], Any] = ...) -> List[_T]: ... + key: Optional[Callable[[_T], Any]] = ...) -> List[_T]: ... def nsmallest(n: int, iterable: Iterable[_T], key: Callable[[_T], Any] = ...) -> List[_T]: ...