Skip to content

Commit 9e23615

Browse files
Update conv.jl
added changes for test
1 parent b36d229 commit 9e23615

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

src/layers/conv.jl

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1170,6 +1170,8 @@ function (l::GMMConv)(g::GNNGraph, x::AbstractMatrix, e::AbstractMatrix)
11701170

11711171
return m
11721172
end
1173+
1174+
(l::GMMConv)(g::GNNGraph) = GNNGraph(g, ndata=l(g, node_features(g), edge_features(g)))
11731175

11741176
function Base.show(io::IO, l::GMMConv)
11751177
(nin, ein), out = l.ch

0 commit comments

Comments
 (0)