From 9331f1ec46ceafd8aeffbbb655e3ec9acee3acdb Mon Sep 17 00:00:00 2001 From: "martha.frysztacki" Date: Thu, 24 Sep 2020 17:46:20 +0200 Subject: [PATCH] cluster_network: The 'squeeze' parameter is deprecated and will be removed in a future release; works just fine without it --- scripts/cluster_network.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/scripts/cluster_network.py b/scripts/cluster_network.py index 6a8e526b..ce1e4319 100644 --- a/scripts/cluster_network.py +++ b/scripts/cluster_network.py @@ -257,7 +257,7 @@ def busmap_for_n_clusters(n, n_clusters, solver_name, focus_weights=None, algori else: raise ValueError(f"`algorithm` must be one of 'kmeans', 'spectral' or 'louvain'. Is {algorithm}.") - return (n.buses.groupby(['country', 'sub_network'], group_keys=False, squeeze=True) + return (n.buses.groupby(['country', 'sub_network'], group_keys=False) .apply(busmap_for_country).rename('busmap')) def plot_busmap_for_n_clusters(n, n_clusters=50):