diff --git a/ash_model/classes/undirected.py b/ash_model/classes/undirected.py index d7dfba9..1633732 100644 --- a/ash_model/classes/undirected.py +++ b/ash_model/classes/undirected.py @@ -387,8 +387,9 @@ def get_neighbors( for s in star: nodes = self.get_hyperedge_nodes(s) res.extend(nodes) - - return set(res) + res = set(res) + res.discard(node) + return res def get_degree(self, node: int, hyperedge_size: int = None, tid: int = None) -> int: """