neighbors count fix

GiulioRossetti · Sep 25, 2024 · 1106b60 · 1106b60
1 parent 9d8ba6d
commit 1106b60
Showing 1 changed file with 3 additions and 2 deletions.
diff --git 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:
         """