From 3fd5c5d0e1a49e57c9b5067afe75da1c9c7e63e6 Mon Sep 17 00:00:00 2001 From: Fabian Neumann Date: Sat, 9 Nov 2019 12:58:04 +0100 Subject: [PATCH] cluster: add tolerance to country weight sum L --- 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 e29f6f96..d7f32dca 100644 --- a/scripts/cluster_network.py +++ b/scripts/cluster_network.py @@ -176,7 +176,7 @@ def distribute_clusters(n, n_clusters, focus_weights=None, solver_name=None): logger.warning('Using custom focus weights for determining number of clusters.') - assert L.sum() == 1.0, "Country weights L must sum up to 1.0 when distributing clusters." + assert np.isclose(L.sum(), 1.0, rtol=1e-3), "Country weights L must sum up to 1.0 when distributing clusters. Is {}.".format(L.sum()) m = po.ConcreteModel() def n_bounds(model, *n_id):