@@ -454,20 +454,13 @@ class TableFormatter:
454
454
455
455
show_dimensions : Union [bool , str ]
456
456
formatters : FormattersType
457
+ columns : Index
457
458
458
459
@property
459
460
def is_truncated (self ) -> bool :
460
461
self ._is_truncated : bool
461
462
return self ._is_truncated
462
463
463
- @property
464
- def columns (self ) -> Index :
465
- return self ._columns
466
-
467
- @columns .setter
468
- def columns (self , columns : Index ) -> None :
469
- self ._columns : Index = columns
470
-
471
464
@property
472
465
def should_show_dimensions (self ) -> bool :
473
466
return self .show_dimensions is True or (
@@ -582,7 +575,6 @@ def __init__(
582
575
self .na_rep = na_rep
583
576
self .decimal = decimal
584
577
self .col_space = self ._initialize_colspace (col_space )
585
-
586
578
self .header = header
587
579
self .index = index
588
580
self .line_width = line_width
@@ -595,10 +587,7 @@ def __init__(
595
587
self .justify = justify # type: ignore[assignment]
596
588
self .bold_rows = bold_rows
597
589
self .escape = escape
598
-
599
- # Ignoring error:
600
- # expression has type "Optional[Sequence[str]]", variable has type "Index"
601
- self .columns = columns # type: ignore[assignment]
590
+ self .columns = self ._initialize_columns (columns )
602
591
603
592
self ._truncate ()
604
593
self .adj = get_adjustment ()
@@ -632,18 +621,13 @@ def justify(self, justify: Optional[str]) -> None:
632
621
else :
633
622
self ._justify = justify
634
623
635
- @property
636
- def columns (self ) -> Index :
637
- return self ._columns
638
-
639
- @columns .setter
640
- def columns (self , columns : Optional [Sequence [str ]]) -> None :
624
+ def _initialize_columns (self , columns : Optional [Sequence [str ]]) -> Index :
641
625
if columns is not None :
642
- self ._columns = ensure_index (columns )
643
- self .frame = self .frame [self ._columns ]
626
+ cols = ensure_index (columns )
627
+ self .frame = self .frame [cols ]
628
+ return cols
644
629
else :
645
- self ._columns = self .frame .columns
646
- assert self ._columns is not None
630
+ return self .frame .columns
647
631
648
632
def _initialize_colspace (
649
633
self , col_space : Optional [ColspaceArgType ]
0 commit comments