From 7c23ef08c5a1b42017dc625e0cc20d555aee9720 Mon Sep 17 00:00:00 2001 From: Thomas Kouroughli Date: Thu, 18 Jan 2024 17:04:22 +0000 Subject: [PATCH] :bug: fix key_error in validation plots --- scripts/plot_validation_electricity_production.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/scripts/plot_validation_electricity_production.py b/scripts/plot_validation_electricity_production.py index 5c5569d0..e95a8126 100644 --- a/scripts/plot_validation_electricity_production.py +++ b/scripts/plot_validation_electricity_production.py @@ -45,6 +45,12 @@ if __name__ == "__main__": header=[0, 1], parse_dates=True, ) + subset_technologies = ["Geothermal", "Nuclear", "Biomass", "Lignite", "Oil", "Coal"] + lowercase_technologies = [ + technology.lower() if technology in subset_technologies else technology + for technology in historic.columns.levels[1] + ] + historic.columns = historic.columns.set_levels(lowercase_technologies, level=1) colors = n.carriers.set_index("nice_name").color.where( lambda s: s != "", "lightgrey"