improve type annotation

Author: Joshix-1

typed_stream/_impl/_iteration_utils.py

@@ -14,7 +14,7 @@
 
 from ..streamable import Streamable
 from ._types import ClassWithCleanUp, IteratorProxy, PrettyRepr
-from ._typing import override
+from ._typing import Self, override
 from ._utils import (
     FunctionWrapperIgnoringArgs,
     IndexValueTuple,
@@ -280,7 +280,7 @@ def __init__(
         self.iterator = iter(iterable)
 
     @override
-    def __iter__(self: V) -> V:
+    def __iter__(self) -> Self:
         """Return self."""
         return self
 

typed_stream/_impl/_lazy_file_iterators.py

@@ -12,7 +12,7 @@
 from typing import IO, AnyStr, TextIO, TypeGuard, TypeVar, overload
 
 from ._types import Closeable, PathLikeType, PrettyRepr
-from ._typing import override
+from ._typing import Self, override
 
 __all__ = (
     "LazyFileIterator",
@@ -86,7 +86,7 @@ def __init__(
         self._file_object = None
 
     @override
-    def __iter__(self) -> Iterator[AnyStr]:
+    def __iter__(self) -> Self:
         """Return self."""
         return self
 

typed_stream/_impl/_types.py

@@ -12,7 +12,7 @@
 from os import PathLike
 from typing import Generic, Protocol, TypeAlias, TypeGuard, TypeVar
 
-from ._typing import override
+from ._typing import Self, override
 
 __all__ = (
     "ClassWithCleanUp",
@@ -166,7 +166,7 @@ def __init__(self, iterable: Iterable[T]) -> None:
         self._iterator = iter(iterable)
 
     @override
-    def __iter__(self) -> Iterator[V]:
+    def __iter__(self) -> Self:
         """Return self."""
         return self