Update graph_utilities.py

ImperialCollegeLondon · Mar 25, 2024 · 3088112 · 3088112
1 parent d3a534a
commit 3088112
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/swmmanywhere/graph_utilities.py b/swmmanywhere/graph_utilities.py
@@ -437,7 +437,10 @@ def __call__(self, G: nx.Graph, **kwargs) -> nx.Graph:
             start_point_edge = data['geometry'].coords[0]
             end_point_node = (G.nodes[v]['x'], G.nodes[v]['y'])
             end_point_edge = data['geometry'].coords[-1]
-            if (start_point_edge != start_point_node) | \
+            if (start_point_edge == end_point_node) & \
+                    (end_point_edge == start_point_node):
+                data['geometry'] = data['geometry'].reverse()
+            elif (start_point_edge != start_point_node) | \
                     (end_point_edge != end_point_node):
                 data['geometry'] = shapely.LineString([start_point_node,
                                                        end_point_node])