Merge pull request #60 from yuehhua/fix

Fix reference bug

Author: yuehhua

src/graph/simplegraphs.jl

@@ -11,7 +11,7 @@ end
 function GCNConv(g::AbstractSimpleGraph, ch::Pair{<:Integer,<:Integer}, σ = identity;
                  init = glorot_uniform, T::DataType=Float32, bias::Bool=true)
     b = bias ? init(ch[2]) : zeros(T, ch[2])
-    fg = FeaturedGraph(Ref(g), Ref(nothing))
+    fg = FeaturedGraph(g, nothing)
     GCNConv(init(ch[2], ch[1]), b, σ, fg)
 end
 

src/graph/weightedgraphs.jl

@@ -10,7 +10,7 @@ end
 function GCNConv(g::AbstractSimpleWeightedGraph, ch::Pair{<:Integer,<:Integer}, σ = identity;
                  init = glorot_uniform, T::DataType=Float32, bias::Bool=true)
     b = bias ? init(ch[2]) : zeros(T, ch[2])
-    fg = FeaturedGraph(Ref(g), Ref(nothing))
+    fg = FeaturedGraph(g, nothing)
     GCNConv(init(ch[2], ch[1]), b, σ, fg)
 end
 

test/graph/metagraphs.jl

@@ -55,12 +55,12 @@ dg = MetaDiGraph(sg)
         gc = GCNConv(ug, in_channel=>out_channel)
         @test size(gc.weight) == (out_channel, in_channel)
         @test size(gc.bias) == (out_channel,)
-        @test graph(gc.graph)[] === ug.graph
+        @test graph(gc.graph) === ug.graph
 
         gc = GCNConv(dg, in_channel=>out_channel)
         @test size(gc.weight) == (out_channel, in_channel)
         @test size(gc.bias) == (out_channel,)
-        @test graph(gc.graph)[] === dg.graph
+        @test graph(gc.graph) === dg.graph
     end
 
     @testset "ChebConv" begin

test/graph/simplegraphs.jl

@@ -63,7 +63,7 @@ el_dg = Vector{Int64}[[3, 6], [3, 5], [4, 5], [], [], []]
         gc = GCNConv(ug, in_channel=>out_channel)
         @test size(gc.weight) == (out_channel, in_channel)
         @test size(gc.bias) == (out_channel,)
-        @test graph(gc.graph)[] === ug
+        @test graph(gc.graph) === ug
     end
 
     @testset "ChebConv" begin

test/graph/weightedgraphs.jl

@@ -63,7 +63,7 @@ el_dg = Vector{Int64}[[3, 6], [3, 5], [4, 5], [], [], []]
         gc = GCNConv(ug, in_channel=>out_channel)
         @test size(gc.weight) == (out_channel, in_channel)
         @test size(gc.bias) == (out_channel,)
-        @test graph(gc.graph)[] === ug
+        @test graph(gc.graph) === ug
     end
 
     @testset "ChebConv" begin