Define generators as Iterator

kakaobrain · Jun 24, 2019 · c7749c2 · c7749c2
1 parent fe945da
commit c7749c2
Showing 1 changed file with 9 additions and 9 deletions.
diff --git a/stubs/torch/nn/__init__.pyi b/stubs/torch/nn/__init__.pyi
@@ -40,15 +40,15 @@ class Module:
     def register_forward_pre_hook(self, hook: __Hook1) -> __RemovableHandle: ...
     def register_forward_hook(self, hook: __Hook2) -> __RemovableHandle: ...
 
-    def parameters(self, recurse: bool = True) -> Iterable[Parameter]: ...
-    def buffers(self, recurse: bool = True) -> Iterable[Tensor]: ...
-    def modules(self) -> Iterable[Module]: ...
-    def children(self) -> Iterable[Module]: ...
-
-    def named_parameters(self, prefix: str = '', recurse: bool = True) -> Iterable[Tuple[str, Parameter]]: ...
-    def named_buffers(self, prefix: str = '', recurse: bool = True) -> Iterable[Tuple[str, Tensor]]: ...
-    def named_modules(self, memo: None = None, prefix: str = '') -> Iterable[Tuple[str, Module]]: ...
-    def named_children(self) -> Iterable[Tuple[str, Module]]: ...
+    def parameters(self, recurse: bool = True) -> Iterator[Parameter]: ...
+    def buffers(self, recurse: bool = True) -> Iterator[Tensor]: ...
+    def modules(self) -> Iterator[Module]: ...
+    def children(self) -> Iterator[Module]: ...
+
+    def named_parameters(self, prefix: str = '', recurse: bool = True) -> Iterator[Tuple[str, Parameter]]: ...
+    def named_buffers(self, prefix: str = '', recurse: bool = True) -> Iterator[Tuple[str, Tensor]]: ...
+    def named_modules(self, memo: None = None, prefix: str = '') -> Iterator[Tuple[str, Module]]: ...
+    def named_children(self) -> Iterator[Tuple[str, Module]]: ...
 
     def add_module(self, name: str, module: Module) -> None: ...