File tree Expand file tree Collapse file tree 1 file changed +5
-3
lines changed Expand file tree Collapse file tree 1 file changed +5
-3
lines changed Original file line number Diff line number Diff line change 39
39
Index ,
40
40
MultiIndex ,
41
41
)
42
+ from pandas .core .indexes .frozen import FrozenList
42
43
from pandas .core .series import Series
43
44
from pandas .core .sorting import (
44
45
compress_group_index ,
@@ -316,15 +317,16 @@ def get_new_columns(self, value_columns: Index | None):
316
317
stride = len (self .removed_level ) + self .lift
317
318
width = len (value_columns )
318
319
propagator = np .repeat (np .arange (width ), stride )
320
+
321
+ new_levels : FrozenList | list [Index ]
322
+
319
323
if isinstance (value_columns , MultiIndex ):
320
324
new_levels = value_columns .levels + (self .removed_level_full ,)
321
325
new_names = value_columns .names + (self .removed_name ,)
322
326
323
327
new_codes = [lab .take (propagator ) for lab in value_columns .codes ]
324
328
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 = [
328
330
value_columns ,
329
331
self .removed_level_full ,
330
332
]
You can’t perform that action at this time.
0 commit comments