fix: add return statement

simulate-digital-rail · Apr 18, 2023 · 35afe3b · 35afe3b
1 parent d5e7a59
commit 35afe3b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/yaramo/node.py b/yaramo/node.py
@@ -94,7 +94,7 @@ def remove_edge_to_node(self, node: "Node"):
 
     def get_edge_to_node(self, node):
         """Returns the edge to the given neighbor node."""
-        next(edge for edge in self.connected_edges if edge.get_opposite_node(self) == node)
+        return next(edge for edge in self.connected_edges if edge.get_opposite_node(self) == node)
 
     def get_possible_followers(self, source):
         """Returns the Nodes that could follow (head, left, right) when comming from a source Node connected to this Node."""