Skip to content

Commit 15d55d1

Browse files
Update test/layers/conv.jl
Co-authored-by: Carlo Lucibello <[email protected]>
1 parent 86d1d08 commit 15d55d1

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

test/layers/conv.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -268,7 +268,7 @@
268268
K = 5
269269
l = GMMConv((in_channel, ein_channel )=> out_channel, K=K)
270270
for g in test_graphs
271-
g = GNNGraph(g, edata=rand(ein_channel, g.num_edges))
271+
g = GNNGraph(g, edata=rand(Float32, ein_channel, g.num_edges))
272272
test_layer(l, g, rtol=RTOL_HIGH, outsize = (out_channel, g.num_nodes))
273273
end
274274
end

0 commit comments

Comments
 (0)