Fix type annotation of LineageHandler to accept None as distance …

…argument
althonos · Jul 18, 2020 · a8f731a · a8f731a
1 parent bc4cf7b
commit a8f731a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pronto/logic/lineage.py b/pronto/logic/lineage.py
@@ -44,11 +44,11 @@ def __eq__(self, other: object) -> bool:
 
 class LineageHandler(Iterable["Term"]):
 
-    def __init__(self, term: "Term", distance: int, with_self: Optional[int]):
+    def __init__(self, term: "Term", distance: Optional[int], with_self: Optional[int]):
         self.term = term
         self.distance = distance
         self.with_self = with_self
-        # API compatibilty with previous iterator
+        # TODO: API compatibilty with previous iterator (remove for v3.0.0)
         self._it = None
 
     def __next__(self):