@@ -380,8 +380,10 @@ def test_multiple_windows(df_time):
380
380
X = df_time .copy ()
381
381
num_vars = ["ambient_temp" , "module_temp" , "irradiation" ]
382
382
tmp = X [num_vars ].rolling (2 ).agg (["sum" , "mean" ]).shift (periods = 15 , freq = "min" )
383
+ tmp .columns = tmp .columns .droplevel ()
383
384
X_tr = X .merge (tmp , left_index = True , right_index = True , how = "left" )
384
385
tmp = X [num_vars ].rolling (3 ).agg (["sum" , "mean" ]).shift (periods = 15 , freq = "min" )
386
+ tmp .columns = tmp .columns .droplevel ()
385
387
X_tr = X_tr .merge (tmp , left_index = True , right_index = True , how = "left" )
386
388
X_tr .columns = transformer .get_feature_names_out ()
387
389
@@ -404,13 +406,15 @@ def test_multiple_windows(df_time):
404
406
.agg (["sum" , "mean" ])
405
407
.shift (freq = "30min" )
406
408
)
409
+ tmp .columns = tmp .columns .droplevel ()
407
410
X_tr = X .merge (tmp , left_index = True , right_index = True , how = "left" )
408
411
tmp = (
409
412
X [["ambient_temp" , "irradiation" ]]
410
413
.rolling (3 )
411
414
.agg (["sum" , "mean" ])
412
415
.shift (freq = "30min" )
413
416
)
417
+ tmp .columns = tmp .columns .droplevel ()
414
418
X_tr = X_tr .merge (tmp , left_index = True , right_index = True , how = "left" )
415
419
X_tr .columns = transformer .get_feature_names_out ()
416
420
0 commit comments