From 676446569dddaca76500178599b05045ae2c3e6a Mon Sep 17 00:00:00 2001
From: Serious-senpai <57554044+Serious-senpai@users.noreply.github.com>
Date: Sat, 11 Nov 2023 21:23:11 +0700
Subject: [PATCH] Provide more precise type hints for itertools.permutations
 (#11019)

---
 stdlib/itertools.pyi | 15 ++++++++++++---
 1 file changed, 12 insertions(+), 3 deletions(-)

diff --git a/stdlib/itertools.pyi b/stdlib/itertools.pyi
index ffa8e1939..55f9c92d8 100644
--- a/stdlib/itertools.pyi
+++ b/stdlib/itertools.pyi
@@ -241,10 +241,19 @@ class product(Iterator[_T_co]):
     def __iter__(self) -> Self: ...
     def __next__(self) -> _T_co: ...
 
-class permutations(Iterator[tuple[_T, ...]], Generic[_T]):
-    def __init__(self, iterable: Iterable[_T], r: int | None = ...) -> None: ...
+class permutations(Iterator[_T_co]):
+    @overload
+    def __new__(cls, iterable: Iterable[_T], r: Literal[2]) -> permutations[tuple[_T, _T]]: ...
+    @overload
+    def __new__(cls, iterable: Iterable[_T], r: Literal[3]) -> permutations[tuple[_T, _T, _T]]: ...
+    @overload
+    def __new__(cls, iterable: Iterable[_T], r: Literal[4]) -> permutations[tuple[_T, _T, _T, _T]]: ...
+    @overload
+    def __new__(cls, iterable: Iterable[_T], r: Literal[5]) -> permutations[tuple[_T, _T, _T, _T, _T]]: ...
+    @overload
+    def __new__(cls, iterable: Iterable[_T], r: int | None = ...) -> permutations[tuple[_T, ...]]: ...
     def __iter__(self) -> Self: ...
-    def __next__(self) -> tuple[_T, ...]: ...
+    def __next__(self) -> _T_co: ...
 
 class combinations(Iterator[_T_co]):
     @overload