Update test/layers/conv.jl

itsmohitanand · CarloLucibello · web-flow · commit 3d141934a5a7 · 2022-03-22T08:34:26.000+01:00
Co-authored-by: Carlo Lucibello &lt;carlo.lucibello@gmail.com&gt;
diff --git a/test/layers/conv.jl b/test/layers/conv.jl
@@ -269,7 +269,7 @@
         l = GMMConv(in_channel => out_channel, K=K, e_dim=e_dim)
         for g in test_graphs
             g = GNNGraph(g, edata=rand(e_dim, g.num_edges))
-            test_layer(l, g, rtol=RTOL_HIGH, outsize = (out_channel, g.num_edges))
+            test_layer(l, g, rtol=RTOL_HIGH, outsize = (out_channel, g.num_nodes))
         end
     end
 end
