From 04d88c8f2ffc16e010b2f26dbdc206eac0153ef9 Mon Sep 17 00:00:00 2001 From: Brandon Rodriguez <brodriguez8774@gmail.com> Date: Sun, 17 Nov 2019 05:34:52 -0500 Subject: [PATCH] Add handling for directed classes to "disconnect_nodes" edge function --- resources/graphs/basic_graph/components.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/resources/graphs/basic_graph/components.py b/resources/graphs/basic_graph/components.py index cbeec67..e12c82d 100644 --- a/resources/graphs/basic_graph/components.py +++ b/resources/graphs/basic_graph/components.py @@ -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 -- GitLab