Skip to content

Commit 51cbd77

Browse files
committed
fix mypy errors
1 parent 7854a1c commit 51cbd77

File tree

4 files changed

+13
-9
lines changed

4 files changed

+13
-9
lines changed

pandas/core/indexes/datetimelike.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -339,7 +339,7 @@ def argmax(self, axis=None, skipna=True, *args, **kwargs):
339339
# Rendering Methods
340340

341341
def format(
342-
self, name: bool = False, formatter=None, date_format=None, na_rep="NaT"
342+
self, name: bool = False, formatter=None, na_rep="NaT", date_format=None
343343
) -> List[str]:
344344
"""
345345
Render a string representation of the Index.

pandas/core/indexes/multi.py

+10-6
Original file line numberDiff line numberDiff line change
@@ -1231,13 +1231,17 @@ def _format_native_types(self, na_rep="nan", **kwargs):
12311231

12321232
def format(
12331233
self,
1234+
name=None,
1235+
formatter=None,
1236+
na_rep=None,
1237+
names=False,
12341238
space=2,
12351239
sparsify=None,
12361240
adjoin=True,
1237-
names=False,
1238-
na_rep=None,
1239-
formatter=None,
12401241
) -> list:
1242+
if name is not None:
1243+
names = name
1244+
12411245
if len(self) == 0:
12421246
return []
12431247

@@ -1265,13 +1269,13 @@ def format(
12651269
stringified_levels.append(formatted)
12661270

12671271
result_levels = []
1268-
for lev, name in zip(stringified_levels, self.names):
1272+
for lev, lev_name in zip(stringified_levels, self.names):
12691273
level = []
12701274

12711275
if names:
12721276
level.append(
1273-
pprint_thing(name, escape_chars=("\t", "\r", "\n"))
1274-
if name is not None
1277+
pprint_thing(lev_name, escape_chars=("\t", "\r", "\n"))
1278+
if lev_name is not None
12751279
else ""
12761280
)
12771281

pandas/io/formats/format.py

+1-2
Original file line numberDiff line numberDiff line change
@@ -330,9 +330,8 @@ def _get_footer(self) -> str:
330330

331331
def _get_formatted_index(self) -> Tuple[List[str], bool]:
332332
index = self.tr_series.index
333-
is_multi = isinstance(index, MultiIndex)
334333

335-
if is_multi:
334+
if isinstance(index, MultiIndex):
336335
have_header = any(name for name in index.names)
337336
fmt_index = index.format(names=True)
338337
else:

pandas/io/formats/html.py

+1
Original file line numberDiff line numberDiff line change
@@ -442,6 +442,7 @@ def _write_hierarchical_rows(
442442
frame = self.fmt.tr_frame
443443
nrows = len(frame)
444444

445+
assert isinstance(frame.index, MultiIndex)
445446
idx_values = frame.index.format(sparsify=False, adjoin=False, names=False)
446447
idx_values = list(zip(*idx_values))
447448

0 commit comments

Comments
 (0)