adjust typing, clean up imports

Author: maxfischer2781

asyncstdlib/itertools.py

@@ -32,7 +32,6 @@
     zip,
     enumerate as aenumerate,
     iter as aiter,
-    tuple as atuple,
 )
 
 S = TypeVar("S")

asyncstdlib/itertools.pyi

@@ -32,27 +32,33 @@ def accumulate(
     initial: T1,
 ) -> AsyncIterator[T1]: ...
 @overload
-def batched(iterable: AnyIterable[T], n: Literal[1]) -> AsyncIterator[tuple[T]]: ...
+def batched(
+    iterable: AnyIterable[T], n: Literal[1], strict: bool = ...
+) -> AsyncIterator[tuple[T]]: ...
 @overload
-def batched(iterable: AnyIterable[T], n: Literal[2]) -> AsyncIterator[tuple[T, T]]: ...
+def batched(
+    iterable: AnyIterable[T], n: Literal[2], strict: bool = ...
+) -> AsyncIterator[tuple[T, T]]: ...
 @overload
 def batched(
-    iterable: AnyIterable[T], n: Literal[3]
+    iterable: AnyIterable[T], n: Literal[3], strict: bool = ...
 ) -> AsyncIterator[tuple[T, T, T]]: ...
 @overload
 def batched(
-    iterable: AnyIterable[T], n: Literal[4]
+    iterable: AnyIterable[T], n: Literal[4], strict: bool = ...
 ) -> AsyncIterator[tuple[T, T, T, T]]: ...
 @overload
 def batched(
-    iterable: AnyIterable[T], n: Literal[5]
+    iterable: AnyIterable[T], n: Literal[5], strict: bool = ...
 ) -> AsyncIterator[tuple[T, T, T, T, T]]: ...
 @overload
 def batched(
-    iterable: AnyIterable[T], n: Literal[6]
+    iterable: AnyIterable[T], n: Literal[6], strict: bool = ...
 ) -> AsyncIterator[tuple[T, T, T, T, T, T]]: ...
 @overload
-def batched(iterable: AnyIterable[T], n: int) -> AsyncIterator[tuple[T, ...]]: ...
+def batched(
+    iterable: AnyIterable[T], n: int, strict: bool = ...
+) -> AsyncIterator[tuple[T, ...]]: ...
 
 class chain(AsyncIterator[T]):
     __slots__: tuple[str, ...]