Skip to content
Snippets Groups Projects
Commit 04d88c8f authored by Brandon Rodriguez's avatar Brandon Rodriguez
Browse files

Add handling for directed classes to "disconnect_nodes" edge function

parent 8e317fbc
Branches
No related merge requests found
......@@ -211,6 +211,10 @@ class BasicEdge():
# Get connected nodes.
try:
node_1, node_2 = self.get_nodes()
# Handle for dict, if classes are of Directed type.
if isinstance(node_1, str):
node_1, node_2 = self.get_nodes().values()
except ValueError:
# No nodes connected. Return None.
return None
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment