Fixed incorrect type hints.

marko-k0 · Jun 13, 2018 · 07886c1 · 07886c1
1 parent aef4799
commit 07886c1
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 4 deletions.
diff --git a/iota/crypto/addresses.py b/iota/crypto/addresses.py
@@ -2,7 +2,7 @@
 from __future__ import absolute_import, division, print_function, \
     unicode_literals
 
-from typing import Generator, Iterable, List, MutableSequence
+from typing import Generator, Iterable, List
 
 from iota import Address, TRITS_PER_TRYTE, TrytesCompatible
 from iota.crypto.kerl import Kerl
@@ -132,7 +132,7 @@ def get_addresses(self, start, count=1, step=1):
         return addresses
 
     def create_iterator(self, start=0, step=1):
-        # type: (int, int) -> Generator[Address]
+        # type: (int, int) -> Generator[Address, None, None]
         """
         Creates an iterator that can be used to progressively generate new
         addresses.

diff --git a/iota/crypto/signing.py b/iota/crypto/signing.py
@@ -15,6 +15,7 @@
 
 __all__ = [
     'KeyGenerator',
+    'KeyIterator',
     'SignatureFragmentGenerator',
     'validate_signature_fragments',
 ]

diff --git a/iota/transaction/validator.py b/iota/transaction/validator.py
@@ -68,7 +68,7 @@ def is_valid(self):
     return not self._errors
 
   def _create_validator(self):
-    # type: () -> Generator[Text]
+    # type: () -> Generator[Text, None, None]
     """
     Creates a generator that does all the work.
     """

diff --git a/iota/types.py b/iota/types.py
@@ -321,7 +321,7 @@ def __len__(self):
     return len(self._trytes)
 
   def __iter__(self):
-    # type: () -> Generator[binary_type]
+    # type: () -> Generator[binary_type, None, None]
     # :see: http://stackoverflow.com/a/14267935/
     return (binary_type(self._trytes[i:i + 1]) for i in range(len(self)))