build_heat_totals: allow subset of countries

2024-03-14 18:47:09 +01:00 · 2024-03-14 18:47:09 +01:00 · f74e2f1bb7
commit f74e2f1bb7
parent 02fc334a2e
1 changed files with 1 additions and 1 deletions
--- a/scripts/build_heat_totals.py
+++ b/scripts/build_heat_totals.py
@ -16,7 +16,7 @@ idx = pd.IndexSlice

 def approximate_heat_demand(energy_totals, hdd):

-    countries = hdd.columns
+    countries = hdd.columns.intersection(energy_totals.index.levels[0])

    demands = {}