Skip to content

Commit ddfc9a2

Browse files
simonjayhawkinsWillAyd
authored andcommitted
TYPING: --disallow-any-expr for HTMLFormatter.__init__ (#28140)
1 parent 294a22c commit ddfc9a2

File tree

3 files changed

+11
-8
lines changed

3 files changed

+11
-8
lines changed

pandas/io/formats/format.py

+4-2
Original file line numberDiff line numberDiff line change
@@ -549,7 +549,8 @@ def __init__(
549549
decimal: str = ".",
550550
table_id: Optional[str] = None,
551551
render_links: bool = False,
552-
**kwds
552+
bold_rows: bool = False,
553+
escape: bool = True,
553554
):
554555
self.frame = frame
555556
self.show_index_names = index_names
@@ -580,7 +581,8 @@ def __init__(
580581
else:
581582
self.justify = justify
582583

583-
self.kwds = kwds
584+
self.bold_rows = bold_rows
585+
self.escape = escape
584586

585587
if columns is not None:
586588
self.columns = ensure_index(columns)

pandas/io/formats/html.py

+4-4
Original file line numberDiff line numberDiff line change
@@ -37,7 +37,7 @@ class HTMLFormatter(TableFormatter):
3737
def __init__(
3838
self,
3939
formatter: DataFrameFormatter,
40-
classes: Optional[Union[str, List, Tuple]] = None,
40+
classes: Optional[Union[str, List[str], Tuple[str, ...]]] = None,
4141
border: Optional[int] = None,
4242
) -> None:
4343
self.fmt = formatter
@@ -46,11 +46,11 @@ def __init__(
4646
self.frame = self.fmt.frame
4747
self.columns = self.fmt.tr_frame.columns
4848
self.elements = [] # type: List[str]
49-
self.bold_rows = self.fmt.kwds.get("bold_rows", False)
50-
self.escape = self.fmt.kwds.get("escape", True)
49+
self.bold_rows = self.fmt.bold_rows
50+
self.escape = self.fmt.escape
5151
self.show_dimensions = self.fmt.show_dimensions
5252
if border is None:
53-
border = get_option("display.html.border")
53+
border = cast(int, get_option("display.html.border"))
5454
self.border = border
5555
self.table_id = self.fmt.table_id
5656
self.render_links = self.fmt.render_links

pandas/io/formats/latex.py

+3-2
Original file line numberDiff line numberDiff line change
@@ -39,12 +39,13 @@ def __init__(
3939
):
4040
self.fmt = formatter
4141
self.frame = self.fmt.frame
42-
self.bold_rows = self.fmt.kwds.get("bold_rows", False)
42+
self.bold_rows = self.fmt.bold_rows
4343
self.column_format = column_format
4444
self.longtable = longtable
4545
self.multicolumn = multicolumn
4646
self.multicolumn_format = multicolumn_format
4747
self.multirow = multirow
48+
self.escape = self.fmt.escape
4849

4950
def write_result(self, buf: IO[str]) -> None:
5051
"""
@@ -142,7 +143,7 @@ def pad_empties(x):
142143
buf.write("\\endfoot\n\n")
143144
buf.write("\\bottomrule\n")
144145
buf.write("\\endlastfoot\n")
145-
if self.fmt.kwds.get("escape", True):
146+
if self.escape:
146147
# escape backslashes first
147148
crow = [
148149
(

0 commit comments

Comments
 (0)