Skip to content

Commit 0d24bfd

Browse files
fix
1 parent abf0f26 commit 0d24bfd

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

src/gnngraph.jl

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -287,7 +287,7 @@ function normalized_adjacency(g::GNNGraph, T::DataType=Float32;
287287
add_self_loops::Bool=false, dir::Symbol=:out)
288288
A = adjacency_matrix(g, T; dir=dir)
289289
if add_self_loops
290-
A += I
290+
A = A + I
291291
end
292292
degs = vec(sum(A; dims=2))
293293
inv_sqrtD = Diagonal(inv.(sqrt.(degs)))
@@ -345,9 +345,9 @@ function add_self_loops(g::GNNGraph{<:COO_T})
345345
end
346346

347347
function add_self_loops(g::GNNGraph{<:ADJMAT_T})
348-
A = adjaceny_matrix(g)
348+
A = g.graph
349349
@assert g.edata === (;)
350-
A += I
350+
A = A + I
351351
num_edges = g.num_edges + g.num_nodes
352352
GNNGraph(A,
353353
g.num_nodes, num_edges, g.num_graphs,

0 commit comments

Comments
 (0)