Skip to content

Commit caea0da

Browse files
committed
Update get_node to be more efficient
1 parent 227ecbd commit caea0da

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

lib/ramble/ramble/graphs.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -148,9 +148,9 @@ def get_node(self, key):
148148
(ramble.util.graph.GraphNode): Node representing the key
149149
requested. Returns None if the key isn't found.
150150
"""
151-
for node in self.walk():
152-
if node.key == key:
153-
return node
151+
node = self.node_definitions.get(key)
152+
if node is not None and node in self.adj_list:
153+
return node
154154
return None
155155

156156

0 commit comments

Comments
 (0)