Skip to content

Commit 7e3927a

Browse files
TYP: Fix mypy error in MultiIndex.factorize return type
1 parent 632b2d6 commit 7e3927a

File tree

1 file changed

+7
-2
lines changed

1 file changed

+7
-2
lines changed

pandas/core/indexes/multi.py

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1436,7 +1436,7 @@ def f(dtype) -> bool:
14361436
return any(f(level.dtype) for level in self.levels)
14371437

14381438
# Cannot determine type of "memory_usage"
1439-
@doc(Index.memory_usage) # type: ignore[has-type]
1439+
@doc(Index.memory_usage)
14401440
def memory_usage(self, deep: bool = False) -> int:
14411441
# we are overwriting our base class to avoid
14421442
# computing .values here which could materialize
@@ -4030,7 +4030,12 @@ def factorize(
40304030

40314031
if not has_extension_dtypes:
40324032
# Use parent implementation for performance when no extension dtypes
4033-
return super().factorize(sort=sort, use_na_sentinel=use_na_sentinel)
4033+
codes, uniques = super().factorize(
4034+
sort=sort, use_na_sentinel=use_na_sentinel
4035+
)
4036+
4037+
assert isinstance(uniques, MultiIndex)
4038+
return codes, uniques
40344039

40354040
# Custom implementation for extension dtypes (GH#62337)
40364041
return self._factorize_with_extension_dtypes(

0 commit comments

Comments
 (0)