Add rng to randn

Co-authored-by: Carlo Lucibello <[email protected]>
JuliaGraphs · Nov 7, 2024 · 48c5f7a · 48c5f7a
1 parent fee17fd
commit 48c5f7a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/GNNLux/src/layers/basic.jl b/GNNLux/src/layers/basic.jl
@@ -35,7 +35,7 @@ julia> m = GNNChain(GCNConv(2=>5),
                     x -> relu.(x), 
                     Dense(5=>4))
 
-julia> x = randn(Float32, 2, 3);
+julia> x = randn(rng, Float32, 2, 3);
 
 julia> g = rand_graph(3, 6)
 GNNGraph: