Update swmmanywhere/graph_utilities.py

Co-authored-by: Diego Alonso Álvarez <6095790+dalonsoa@users.noreply.github.com>
ImperialCollegeLondon · Apr 25, 2024 · ca84160 · ca84160
1 parent 7212423
commit ca84160
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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