From 1106b60990e4382e2f4b4d502f105bc1d47e7b22 Mon Sep 17 00:00:00 2001 From: andreafailla <77196675+andreafailla@users.noreply.github.com> Date: Wed, 25 Sep 2024 16:21:42 +0200 Subject: [PATCH] neighbors count fix --- ash_model/classes/undirected.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) 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: """