From b1d5f2f8d5c1d94ff2f0a4d1372995f08dd8108d Mon Sep 17 00:00:00 2001
From: Serious-senpai <57554044+Serious-senpai@users.noreply.github.com>
Date: Sat, 25 Nov 2023 00:46:39 +0700
Subject: [PATCH] Provide precise type hints for combinations_with_replacement
 (#11068)

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

diff --git a/stdlib/itertools.pyi b/stdlib/itertools.pyi
index 881fb236b..de49ee3c5 100644
--- a/stdlib/itertools.pyi
+++ b/stdlib/itertools.pyi
@@ -270,10 +270,19 @@ class combinations(Iterator[_T_co]):
     def __iter__(self) -> Self: ...
     def __next__(self) -> _T_co: ...
 
-class combinations_with_replacement(Iterator[tuple[_T, ...]], Generic[_T]):
-    def __init__(self, iterable: Iterable[_T], r: int) -> None: ...
+class combinations_with_replacement(Iterator[_T_co]):
+    @overload
+    def __new__(cls, iterable: Iterable[_T], r: Literal[2]) -> combinations_with_replacement[tuple[_T, _T]]: ...
+    @overload
+    def __new__(cls, iterable: Iterable[_T], r: Literal[3]) -> combinations_with_replacement[tuple[_T, _T, _T]]: ...
+    @overload
+    def __new__(cls, iterable: Iterable[_T], r: Literal[4]) -> combinations_with_replacement[tuple[_T, _T, _T, _T]]: ...
+    @overload
+    def __new__(cls, iterable: Iterable[_T], r: Literal[5]) -> combinations_with_replacement[tuple[_T, _T, _T, _T, _T]]: ...
+    @overload
+    def __new__(cls, iterable: Iterable[_T], r: int) -> combinations_with_replacement[tuple[_T, ...]]: ...
     def __iter__(self) -> Self: ...
-    def __next__(self) -> tuple[_T, ...]: ...
+    def __next__(self) -> _T_co: ...
 
 if sys.version_info >= (3, 10):
     class pairwise(Iterator[_T_co]):