1111import re
1212
1313from collections .abc import Iterable
14- from typing import Optional
14+ from typing import List , Optional , Tuple , Union
1515
1616from tabulate import tabulate
1717
@@ -40,7 +40,7 @@ def __init__(
4040 self ,
4141 sequence : Iterable ,
4242 transform = None ,
43- engine : ExecutionEngine = None ,
43+ engine : Optional [ ExecutionEngine ] = None ,
4444 max_repr_items : Optional [int ] = None ,
4545 no_wrap : Optional [bool ] = None ,
4646 ):
@@ -59,9 +59,13 @@ def __init__(
5959 """
6060 self .engine = engine or ExecutionEngine ()
6161 if isinstance (sequence , Sequence ):
62- self ._max_repr_items = max_repr_items or sequence ._max_repr_items
63- self ._base_sequence = sequence ._base_sequence
64- self ._lineage = Lineage (prior_lineage = sequence ._lineage , engine = engine )
62+ self ._max_repr_items : Optional [int ] = (
63+ max_repr_items or sequence ._max_repr_items
64+ )
65+ self ._base_sequence : Union [Iterable , List , Tuple ] = sequence ._base_sequence
66+ self ._lineage : Lineage = Lineage (
67+ prior_lineage = sequence ._lineage , engine = engine
68+ )
6569 elif isinstance (sequence , (list , tuple )) or is_iterable (sequence ):
6670 self ._max_repr_items = max_repr_items
6771 self ._base_sequence = sequence
0 commit comments