Skip to content

Commit

Permalink
fix mypy errors
Browse files Browse the repository at this point in the history
  • Loading branch information
topper-123 committed Jul 4, 2020
1 parent 7854a1c commit 31c4eb7
Show file tree
Hide file tree
Showing 5 changed files with 14 additions and 10 deletions.
2 changes: 1 addition & 1 deletion pandas/core/indexes/base.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,9 +8,9 @@
Callable,
FrozenSet,
Hashable,
List,
Optional,
Union,
List,
)
import warnings

Expand Down
2 changes: 1 addition & 1 deletion pandas/core/indexes/datetimelike.py
Original file line number Diff line number Diff line change
Expand Up @@ -339,7 +339,7 @@ def argmax(self, axis=None, skipna=True, *args, **kwargs):
# Rendering Methods

def format(
self, name: bool = False, formatter=None, date_format=None, na_rep="NaT"
self, name: bool = False, formatter=None, na_rep="NaT", date_format=None
) -> List[str]:
"""
Render a string representation of the Index.
Expand Down
16 changes: 10 additions & 6 deletions pandas/core/indexes/multi.py
Original file line number Diff line number Diff line change
Expand Up @@ -1231,13 +1231,17 @@ def _format_native_types(self, na_rep="nan", **kwargs):

def format(
self,
name=None,
formatter=None,
na_rep=None,
names=False,
space=2,
sparsify=None,
adjoin=True,
names=False,
na_rep=None,
formatter=None,
) -> list:
if name is not None:
names = name

if len(self) == 0:
return []

Expand Down Expand Up @@ -1265,13 +1269,13 @@ def format(
stringified_levels.append(formatted)

result_levels = []
for lev, name in zip(stringified_levels, self.names):
for lev, lev_name in zip(stringified_levels, self.names):
level = []

if names:
level.append(
pprint_thing(name, escape_chars=("\t", "\r", "\n"))
if name is not None
pprint_thing(lev_name, escape_chars=("\t", "\r", "\n"))
if lev_name is not None
else ""
)

Expand Down
3 changes: 1 addition & 2 deletions pandas/io/formats/format.py
Original file line number Diff line number Diff line change
Expand Up @@ -330,9 +330,8 @@ def _get_footer(self) -> str:

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

if is_multi:
if isinstance(index, MultiIndex):
have_header = any(name for name in index.names)
fmt_index = index.format(names=True)
else:
Expand Down
1 change: 1 addition & 0 deletions pandas/io/formats/html.py
Original file line number Diff line number Diff line change
Expand Up @@ -442,6 +442,7 @@ def _write_hierarchical_rows(
frame = self.fmt.tr_frame
nrows = len(frame)

assert isinstance(frame.index, MultiIndex)
idx_values = frame.index.format(sparsify=False, adjoin=False, names=False)
idx_values = list(zip(*idx_values))

Expand Down

0 comments on commit 31c4eb7

Please sign in to comment.