From ba44b43e3a895ab10deeb72410bf7ebce2652a7b Mon Sep 17 00:00:00 2001 From: Kaan Sancak Date: Fri, 5 Apr 2024 15:59:15 -0400 Subject: [PATCH] Check dgl version for adj --- Node/preprocess_node_data.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/Node/preprocess_node_data.py b/Node/preprocess_node_data.py index 2502c24..c446a8f 100644 --- a/Node/preprocess_node_data.py +++ b/Node/preprocess_node_data.py @@ -124,7 +124,8 @@ def load_data(dataset_str): def eig_dgl_adj_sparse(g, sm=0, lm=0): - A = g.adj(scipy_fmt='csr') + adj_fn = g.adj if version.parse(dgl.__version__) < version.parse('1.1') else g.adj_external + A = adj_fn(scipy_fmt='csr') deg = np.array(A.sum(axis=0)).flatten() D_ = sp.sparse.diags(deg ** -0.5)