Skip to content

Commit 02ec4a8

Browse files
committed
Use droplevel before merging
Merging dfs with different column lelvels has been disallowed ref pandas-dev/pandas#34862
1 parent 26ef3ae commit 02ec4a8

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

tests/test_time_series/test_forecasting/test_window_features.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -380,8 +380,10 @@ def test_multiple_windows(df_time):
380380
X = df_time.copy()
381381
num_vars = ["ambient_temp", "module_temp", "irradiation"]
382382
tmp = X[num_vars].rolling(2).agg(["sum", "mean"]).shift(periods=15, freq="min")
383+
tmp.columns = tmp.columns.droplevel()
383384
X_tr = X.merge(tmp, left_index=True, right_index=True, how="left")
384385
tmp = X[num_vars].rolling(3).agg(["sum", "mean"]).shift(periods=15, freq="min")
386+
tmp.columns = tmp.columns.droplevel()
385387
X_tr = X_tr.merge(tmp, left_index=True, right_index=True, how="left")
386388
X_tr.columns = transformer.get_feature_names_out()
387389

@@ -404,13 +406,15 @@ def test_multiple_windows(df_time):
404406
.agg(["sum", "mean"])
405407
.shift(freq="30min")
406408
)
409+
tmp.columns = tmp.columns.droplevel()
407410
X_tr = X.merge(tmp, left_index=True, right_index=True, how="left")
408411
tmp = (
409412
X[["ambient_temp", "irradiation"]]
410413
.rolling(3)
411414
.agg(["sum", "mean"])
412415
.shift(freq="30min")
413416
)
417+
tmp.columns = tmp.columns.droplevel()
414418
X_tr = X_tr.merge(tmp, left_index=True, right_index=True, how="left")
415419
X_tr.columns = transformer.get_feature_names_out()
416420

0 commit comments

Comments
 (0)