From ca84160b7e40303dec50e0f3717f9ee8b2879eb6 Mon Sep 17 00:00:00 2001 From: barneydobson Date: Thu, 25 Apr 2024 09:08:58 +0100 Subject: [PATCH] Update swmmanywhere/graph_utilities.py MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Co-authored-by: Diego Alonso Álvarez <6095790+dalonsoa@users.noreply.github.com> --- swmmanywhere/graph_utilities.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/swmmanywhere/graph_utilities.py b/swmmanywhere/graph_utilities.py index 58e610e4..523ed089 100644 --- a/swmmanywhere/graph_utilities.py +++ b/swmmanywhere/graph_utilities.py @@ -254,8 +254,8 @@ def __call__(self, edges_to_remove.add((u, v, keys)) elif data.get(omit, None): edges_to_remove.add((u, v, keys)) - for u, v, keys in edges_to_remove: - G.remove_edge(u, v, keys) + for edges in edges_to_remove: + G.remove_edge(*edges) return G @register_graphfcn