From 472a09988101d280b434d5d26947a5408f54cceb Mon Sep 17 00:00:00 2001
From: Fabian Neumann <fabian.neumann@outlook.de>
Date: Fri, 16 Feb 2024 13:30:52 +0100
Subject: [PATCH] add_electricity: set locations and units of electricity buses

---
 scripts/add_electricity.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/scripts/add_electricity.py b/scripts/add_electricity.py
index 614e3330..9ac3638d 100755
--- a/scripts/add_electricity.py
+++ b/scripts/add_electricity.py
@@ -903,5 +903,9 @@ if __name__ == "__main__":
 
     sanitize_carriers(n, snakemake.config)
 
+    # set locations and units of electricity buses
+    n.buses["location"] = n.buses.index
+    n.buses["unit"] = "MWh_el"
+
     n.meta = snakemake.config
     n.export_to_netcdf(snakemake.output[0])