remove looping due to generalisation

AnatomicMaps · May 23, 2024 · f11c7bb · f11c7bb
1 parent ef5574a
commit f11c7bb
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/mapmaker/routing/__init__.py b/mapmaker/routing/__init__.py
@@ -1306,6 +1306,11 @@ def set_direction(upstream_node):
                         if (p[i], p[i+1]) in route_graph.edges:
                             route_graph.remove_edge(p[i], p[i+1])
 
+        # removing self loop due to generalisation
+        for edge in nx.Graph(route_graph):
+            if edge[0] == edge[1] and (edge[0], edge[1]) in route_graph:
+                route_graph.remove_edge(edge[0], edge[1])
+
         if debug:
             return (route_graph, G, connectivity_graph, terminal_graphs)    # type: ignore
         else: