Fixed issue (most likely) introduced when merging PyPSA-Eur with PyPSA-Eur-Sec

This commit is contained in:
Ricardo 2023-06-12 20:40:51 +02:00
parent 778f850a9e
commit b7aa122fda

View File

@ -274,13 +274,13 @@ def add_EQ_constraints(n, o, scaling=1e-1):
float_regex = "[0-9]*\.?[0-9]+"
level = float(re.findall(float_regex, o)[0])
if o[-1] == "c":
ggrouper = n.generators.bus.map(n.buses.country).to_xarray()
lgrouper = n.loads.bus.map(n.buses.country).to_xarray()
sgrouper = n.storage_units.bus.map(n.buses.country).to_xarray()
ggrouper = n.generators.bus.map(n.buses.country)
lgrouper = n.loads.bus.map(n.buses.country)
sgrouper = n.storage_units.bus.map(n.buses.country)
else:
ggrouper = n.generators.bus.to_xarray()
lgrouper = n.loads.bus.to_xarray()
sgrouper = n.storage_units.bus.to_xarray()
ggrouper = n.generators.bus
lgrouper = n.loads.bus
sgrouper = n.storage_units.bus
load = (
n.snapshot_weightings.generators
@ n.loads_t.p_set.groupby(lgrouper, axis=1).sum()
@ -294,7 +294,7 @@ def add_EQ_constraints(n, o, scaling=1e-1):
p = n.model["Generator-p"]
lhs_gen = (
(p * (n.snapshot_weightings.generators * scaling))
.groupby(ggrouper)
.groupby(ggrouper.to_xarray())
.sum()
.sum("snapshot")
)
@ -303,7 +303,7 @@ def add_EQ_constraints(n, o, scaling=1e-1):
spillage = n.model["StorageUnit-spill"]
lhs_spill = (
(spillage * (-n.snapshot_weightings.stores * scaling))
.groupby(sgrouper)
.groupby(sgrouper.to_xarray())
.sum()
.sum("snapshot")
)