From 1e940660d346af5703e69d0a1557f3ee7bf68e80 Mon Sep 17 00:00:00 2001 From: Lukas Winkler Date: Fri, 25 Nov 2022 20:09:35 +0100 Subject: [PATCH] don't add edges to itself --- graph/views.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/graph/views.py b/graph/views.py index 025e637..59a3dd2 100644 --- a/graph/views.py +++ b/graph/views.py @@ -32,9 +32,13 @@ class Graph: }) def add_edge(self, source: GraphModelEl, target: GraphModelEl): + if source == target: + return self.edges.add((source.graphkey, target.graphkey)) def add_edge_str(self, source: str, target: str): + if source == target: + return self.edges.add((source, target)) def prune(self) -> None: