From 60ca4f53a2b2b075586b5d4e90caaaa23c628a13 Mon Sep 17 00:00:00 2001
From: Roman Mogylatov <rmogilatov@gmail.com>
Date: Sun, 7 Feb 2021 14:08:52 -0500
Subject: [PATCH] Improve traverse() typing stubs

---
 src/dependency_injector/containers.pyi | 6 ++----
 src/dependency_injector/providers.pyi  | 2 +-
 2 files changed, 3 insertions(+), 5 deletions(-)

diff --git a/src/dependency_injector/containers.pyi b/src/dependency_injector/containers.pyi
index 2575d9d0..2f154c22 100644
--- a/src/dependency_injector/containers.pyi
+++ b/src/dependency_injector/containers.pyi
@@ -7,9 +7,7 @@ from typing import (
     Union,
     ClassVar,
     Callable as _Callable,
-    Sequence,
     Iterable,
-    Iterator,
     TypeVar,
     Awaitable,
     overload,
@@ -48,10 +46,10 @@ class Container:
     def apply_container_providers_overridings(self) -> None: ...
     def reset_singletons(self) -> None: ...
     @overload
-    def traverse(self, types: Optional[Sequence[TT]] = None) -> Iterator[TT]: ...
+    def traverse(self, types: Optional[Iterable[Type[TT]]] = None) -> _Iterator[TT]: ...
     @classmethod
     @overload
-    def traverse(cls, types: Optional[Sequence[TT]] = None) -> Iterator[TT]: ...
+    def traverse(self, types: Optional[Iterable[Type[TT]]] = None) -> _Iterator[TT]: ...
 
 
 class DynamicContainer(Container): ...
diff --git a/src/dependency_injector/providers.pyi b/src/dependency_injector/providers.pyi
index 2af1b6e6..b05101b0 100644
--- a/src/dependency_injector/providers.pyi
+++ b/src/dependency_injector/providers.pyi
@@ -77,7 +77,7 @@ class Provider(Generic[T]):
     def is_async_mode_undefined(self) -> bool: ...
     @property
     def related(self) -> _Iterator[Provider]: ...
-    def traverse(self, types: Optional[_Iterable[TT]] = None) -> _Iterator[TT]: ...
+    def traverse(self, types: Optional[_Iterable[Type[TT]]] = None) -> _Iterator[TT]: ...
     def _copy_overridings(self, copied: Provider, memo: Optional[_Dict[Any, Any]]) -> None: ...