Updated return type to Iterator[int]

Author: 1227haran

data_structures/stacks/lexicographical_numbers.py

@@ -1,20 +1,20 @@
 from collections.abc import Iterator
 
 
-def lexical_order(max_number: int) -> Iterator[str]:
+def lexical_order(max_number: int) -> Iterator[int]:
     """
     Generate numbers in lexical order from 1 to max_number.
 
-    >>> " ".join(lexical_order(13))
+    >>> " ".join(map(str, lexical_order(13)))
     '1 10 11 12 13 2 3 4 5 6 7 8 9'
     >>> list(lexical_order(1))
-    ['1']
-    >>> " ".join(lexical_order(20))
+    [1]
+    >>> " ".join(map(str, lexical_order(20)))
     '1 10 11 12 13 14 15 16 17 18 19 2 20 3 4 5 6 7 8 9'
-    >>> " ".join(lexical_order(25))
+    >>> " ".join(map(str, lexical_order(25)))
     '1 10 11 12 13 14 15 16 17 18 19 2 20 21 22 23 24 25 3 4 5 6 7 8 9'
     >>> list(lexical_order(12))
-    ['1', '10', '11', '12', '2', '3', '4', '5', '6', '7', '8', '9']
+    [1, 10, 11, 12, 2, 3, 4, 5, 6, 7, 8, 9]
     """
 
     stack = [1]
@@ -24,7 +24,7 @@ def lexical_order(max_number: int) -> Iterator[str]:
         if num > max_number:
             continue
 
-        yield (str(num))
+        yield (num)
         if (num % 10) != 9:
             stack.append(num + 1)