@@ -581,12 +581,7 @@ def __init__(
581
581
self .formatters = self ._initialize_formatters (formatters )
582
582
self .na_rep = na_rep
583
583
self .decimal = decimal
584
-
585
- # Ignoring error
586
- # expression has type "Union[str, int, Sequence[Union[str, int]],
587
- # Mapping[Optional[Hashable], Union[str, int]], None]",
588
- # variable has type "Mapping[Optional[Hashable], Union[str, int]]"
589
- self .col_space = col_space # type: ignore[assignment]
584
+ self .col_space = self ._initialize_colspace (col_space )
590
585
591
586
self .header = header
592
587
self .index = index
@@ -650,33 +645,31 @@ def columns(self, columns: Optional[Sequence[str]]) -> None:
650
645
self ._columns = self .frame .columns
651
646
assert self ._columns is not None
652
647
653
- @property
654
- def col_space (self ) -> ColspaceType :
655
- return self ._col_space
648
+ def _initialize_colspace (
649
+ self , col_space : Optional [ColspaceArgType ]
650
+ ) -> ColspaceType :
651
+ result : ColspaceType
656
652
657
- @col_space .setter
658
- def col_space (self , col_space : Optional [ColspaceArgType ]) -> None :
659
- self ._col_space : ColspaceType
660
653
if col_space is None :
661
- self . _col_space = {}
654
+ result = {}
662
655
elif isinstance (col_space , (int , str )):
663
- self . _col_space = {"" : col_space }
664
- self . _col_space .update ({column : col_space for column in self .frame .columns })
656
+ result = {"" : col_space }
657
+ result .update ({column : col_space for column in self .frame .columns })
665
658
elif isinstance (col_space , Mapping ):
666
659
for column in col_space .keys ():
667
660
if column not in self .frame .columns and column != "" :
668
661
raise ValueError (
669
662
f"Col_space is defined for an unknown column: { column } "
670
663
)
671
- self . _col_space = col_space
664
+ result = col_space
672
665
else :
673
666
if len (self .frame .columns ) != len (col_space ):
674
667
raise ValueError (
675
668
f"Col_space length({ len (col_space )} ) should match "
676
669
f"DataFrame number of columns({ len (self .frame .columns )} )"
677
670
)
678
- self . _col_space = dict (zip (self .frame .columns , col_space ))
679
- assert self . _col_space is not None
671
+ result = dict (zip (self .frame .columns , col_space ))
672
+ return result
680
673
681
674
@property
682
675
def max_rows_displayed (self ) -> int :
0 commit comments