Skip to content

Commit 58417cf

Browse files
author
Kashif Khan
committed
TYP: define new_leveln as FrozenList or List
1 parent a63b165 commit 58417cf

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

pandas/core/reshape/reshape.py

Lines changed: 4 additions & 1 deletion
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,
@@ -317,7 +318,9 @@ def get_new_columns(self, value_columns: Index | None):
317318
width = len(value_columns)
318319
propagator = np.repeat(np.arange(width), stride)
319320
if isinstance(value_columns, MultiIndex):
320-
new_levels = value_columns.levels + (self.removed_level_full,)
321+
new_levels: FrozenList | list[Index] = value_columns.levels + (
322+
self.removed_level_full,
323+
)
321324
new_names = value_columns.names + (self.removed_name,)
322325

323326
new_codes = [lab.take(propagator) for lab in value_columns.codes]

0 commit comments

Comments
 (0)