diff --git a/resources/graphs/basic_graph/components.py b/resources/graphs/basic_graph/components.py index cbeec67d6c8f8599f5ca409469a5967682b9fdfb..e12c82d28b3ff60f6ba9044dbf86fea395f1606e 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