fix: squeeze IDEES dataframe at correct location

2022-04-12 16:11:06 +02:00 · 2022-04-12 16:11:06 +02:00 · c5ca8f4c20
commit c5ca8f4c20
parent d4a82a2a41
1 changed files with 2 additions and 2 deletions
--- a/scripts/build_industry_sector_ratios.py
+++ b/scripts/build_industry_sector_ratios.py
@ -79,10 +79,10 @@ def load_idees_data(sector, country="EU28"):
        index_col=0,
        header=0,
        usecols=usecols,
-    ).squeeze('columns')
+    )

    for k, v in sheets.items():
-        idees[k] = idees.pop(v)
+        idees[k] = idees.pop(v).squeeze()

    return idees