Skip to content

Commit 766bbad

Browse files
simonjayhawkinsKevin D Smith
authored and
Kevin D Smith
committed
TYP: misc cleanup in core\generic.py (pandas-dev#35963)
1 parent c3e0e39 commit 766bbad

File tree

1 file changed

+10
-6
lines changed

1 file changed

+10
-6
lines changed

pandas/core/generic.py

+10-6
Original file line numberDiff line numberDiff line change
@@ -387,7 +387,7 @@ def _get_block_manager_axis(cls, axis: Axis) -> int:
387387
return m - axis
388388
return axis
389389

390-
def _get_axis_resolvers(self, axis: str) -> Dict[str, ABCSeries]:
390+
def _get_axis_resolvers(self, axis: str) -> Dict[str, Union["Series", MultiIndex]]:
391391
# index or columns
392392
axis_index = getattr(self, axis)
393393
d = dict()
@@ -417,10 +417,10 @@ def _get_axis_resolvers(self, axis: str) -> Dict[str, ABCSeries]:
417417
d[axis] = dindex
418418
return d
419419

420-
def _get_index_resolvers(self) -> Dict[str, ABCSeries]:
420+
def _get_index_resolvers(self) -> Dict[str, Union["Series", MultiIndex]]:
421421
from pandas.core.computation.parsing import clean_column_name
422422

423-
d: Dict[str, ABCSeries] = {}
423+
d: Dict[str, Union["Series", MultiIndex]] = {}
424424
for axis_name in self._AXIS_ORDERS:
425425
d.update(self._get_axis_resolvers(axis_name))
426426

@@ -4703,14 +4703,15 @@ def filter(
47034703
return self.reindex(**{name: [r for r in items if r in labels]})
47044704
elif like:
47054705

4706-
def f(x):
4706+
def f(x) -> bool:
4707+
assert like is not None # needed for mypy
47074708
return like in ensure_str(x)
47084709

47094710
values = labels.map(f)
47104711
return self.loc(axis=axis)[values]
47114712
elif regex:
47124713

4713-
def f(x):
4714+
def f(x) -> bool:
47144715
return matcher.search(ensure_str(x)) is not None
47154716

47164717
matcher = re.compile(regex)
@@ -6556,7 +6557,10 @@ def replace(
65566557
regex = True
65576558

65586559
items = list(to_replace.items())
6559-
keys, values = zip(*items) if items else ([], [])
6560+
if items:
6561+
keys, values = zip(*items)
6562+
else:
6563+
keys, values = ([], [])
65606564

65616565
are_mappings = [is_dict_like(v) for v in values]
65626566

0 commit comments

Comments
 (0)