Add rng

Co-authored-by: Carlo Lucibello <[email protected]>
JuliaGraphs · Nov 7, 2024 · 7a6c6e4 · 7a6c6e4
1 parent 9c78897
commit 7a6c6e4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/GNNLux/src/layers/conv.jl b/GNNLux/src/layers/conv.jl
@@ -56,7 +56,7 @@ rng = Random.default_rng()
 s = [1,1,2,3]
 t = [2,3,1,1]
 g = GNNGraph(s, t)
-x = randn(Float32, 3, g.num_nodes)
+x = randn(rng, Float32, 3, g.num_nodes)
 
 # create layer
 l = GCNConv(3 => 5)