Remove isolated nodes from Delwaq network. (#1948)

As these caused off by one issues in the generated Delwaq model. First found in the AGV model.
Deltares · Nov 21, 2024 · 06fe982 · 06fe982
1 parent fe3b789
commit 06fe982
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/python/ribasim/ribasim/delwaq/generate.py b/python/ribasim/ribasim/delwaq/generate.py
@@ -143,6 +143,11 @@ def _setup_graph(nodes, edge, evaporate_mass=True):
                 else:
                     G.add_edge(*edge, id=[edge_id])
 
+    iso = nx.number_of_isolates(G)
+    if iso > 0:
+        logger.debug(f"Found {iso} isolated nodes in the network.")
+        remove_nodes.extend(list(nx.isolates(G)))
+
     for node_id in remove_nodes:
         G.remove_node(node_id)