cluster_network: ensure correct indexing of weights

This commit is contained in:
Fabian 2024-03-25 14:07:02 +01:00
parent d72576aa1a
commit 585f9a32a7

View File

@ -231,7 +231,7 @@ def distribute_clusters(n, n_clusters, focus_weights=None, solver_name="scip"):
.pipe(normed)
)
N = n.buses.groupby(["country", "sub_network"]).size()
N = n.buses.groupby(["country", "sub_network"]).size()[L.index]
assert (
n_clusters >= len(N) and n_clusters <= N.sum()
@ -454,7 +454,7 @@ if __name__ == "__main__":
if "snakemake" not in globals():
from _helpers import mock_snakemake
snakemake = mock_snakemake("cluster_network", simpl="", clusters="5")
snakemake = mock_snakemake("cluster_network", simpl="", clusters="40")
configure_logging(snakemake)
set_scenario_config(snakemake)