Skip to content

Commit fa3d5f1

Browse files
authored
TYP: fix incorrect mypy error in reshape.py (#45127)
1 parent e750c94 commit fa3d5f1

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

pandas/core/reshape/reshape.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -39,6 +39,7 @@
3939
Index,
4040
MultiIndex,
4141
)
42+
from pandas.core.indexes.frozen import FrozenList
4243
from pandas.core.series import Series
4344
from pandas.core.sorting import (
4445
compress_group_index,
@@ -316,15 +317,16 @@ def get_new_columns(self, value_columns: Index | None):
316317
stride = len(self.removed_level) + self.lift
317318
width = len(value_columns)
318319
propagator = np.repeat(np.arange(width), stride)
320+
321+
new_levels: FrozenList | list[Index]
322+
319323
if isinstance(value_columns, MultiIndex):
320324
new_levels = value_columns.levels + (self.removed_level_full,)
321325
new_names = value_columns.names + (self.removed_name,)
322326

323327
new_codes = [lab.take(propagator) for lab in value_columns.codes]
324328
else:
325-
# error: Incompatible types in assignment (expression has type "List[Any]",
326-
# variable has type "FrozenList")
327-
new_levels = [ # type: ignore[assignment]
329+
new_levels = [
328330
value_columns,
329331
self.removed_level_full,
330332
]

0 commit comments

Comments
 (0)