From fd0853eaa170cd418683543bf19c218757a749f7 Mon Sep 17 00:00:00 2001
From: yerbol-akhmetov <yerbol.akhmetov.93@gmail.com>
Date: Wed, 9 Oct 2024 15:59:11 +0500
Subject: [PATCH] fix issue of adding H2 store and make H2 Store in AC buses

---
 scripts/add_extra_components.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/scripts/add_extra_components.py b/scripts/add_extra_components.py
index cab8195df..930243859 100644
--- a/scripts/add_extra_components.py
+++ b/scripts/add_extra_components.py
@@ -105,7 +105,7 @@ def attach_stores(n, costs, config):
     _add_missing_carriers_from_costs(n, costs, carriers)
 
     buses_i = n.buses.query("carrier == 'AC'").index
-    bus_sub_dict = {k: n.buses[k].values for k in ["x", "y", "country"]}
+    bus_sub_dict = {k: n.buses.loc[buses_i, k].values for k in ["x", "y", "country"]}
 
     if "H2" in carriers:
         h2_buses_i = n.madd("Bus", buses_i + " H2", carrier="H2", **bus_sub_dict)