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 3939 Index ,
4040 MultiIndex ,
4141)
42+ from pandas .core .indexes .frozen import FrozenList
4243from pandas .core .series import Series
4344from 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 ]
You can’t perform that action at this time.
0 commit comments