fix

Author: yuehhua

docs/Project.toml

@@ -1,6 +1,7 @@
 [deps]
 Documenter = "e30172f5-a6a5-5a46-863b-614d45cd2de4"
 DocumenterCitations = "daee34ce-89f3-4625-b898-19384cb65244"
+Flux = "587475ba-b771-5e3f-ad9e-33799f191a9c"
 
 [compat]
 Documenter = "0.27"

src/layers/graph_conv.jl

@@ -336,7 +336,7 @@ function (l::GATConv)(fg::AbstractFeaturedGraph)
     GraphSignals.check_num_nodes(fg, X)
     sg = graph(fg)
     @assert ChainRulesCore.ignore_derivatives(() -> GraphSignals.has_all_self_loops(sg)) "a vertex must have self loop (receive a message from itself)."
-    el = to_namedtuple(sg)
+    el = GraphSignals.to_namedtuple(sg)
     _, V, _ = propagate(l, el, nothing, X, nothing, hcat, nothing, nothing)
     return ConcreteFeaturedGraph(fg, nf=V)
 end
@@ -464,7 +464,7 @@ function (l::GATv2Conv)(fg::AbstractFeaturedGraph)
     GraphSignals.check_num_nodes(fg, X)
     sg = graph(fg)
     @assert ChainRulesCore.ignore_derivatives(() -> GraphSignals.has_all_self_loops(sg)) "a vertex must have self loop (receive a message from itself)."
-    el = to_namedtuple(sg)
+    el = GraphSignals.to_namedtuple(sg)
     _, V, _ = propagate(l, el, nothing, X, nothing, hcat, nothing, nothing)
     return ConcreteFeaturedGraph(fg, nf=V)
 end
@@ -540,7 +540,7 @@ update(ggc::GatedGraphConv, m::AbstractArray, x) = m
 function (l::GatedGraphConv)(fg::AbstractFeaturedGraph)
     nf = node_feature(fg)
     GraphSignals.check_num_nodes(fg, nf)
-    V = l(GraphSignals.to_namedtuple(fg), nf)
+    V = l(GraphSignals.GraphSignals.to_namedtuple(fg), nf)
     return ConcreteFeaturedGraph(fg, nf=V)
 end
 

test/layers/group_conv.jl

@@ -28,7 +28,7 @@
             @test size(nf_) == (out_channel, N)
             @test size(pf_) == (pos_dim, N)
 
-            g = Zygote.gradient(() -> sum(node_feature(egnn(fg))), Flux.params(egnn))
+            g = gradient(() -> sum(node_feature(egnn(fg))), Flux.params(egnn))
             @test length(g.grads) == 13
         end
 
@@ -45,7 +45,7 @@
             @test size(nf_) == (out_channel, N)
             @test size(pf_) == (pos_dim, N)
 
-            g = Zygote.gradient(() -> sum(node_feature(egnn(fg))), Flux.params(egnn))
+            g = gradient(() -> sum(node_feature(egnn(fg))), Flux.params(egnn))
             @test length(g.grads) == 13
         end
 
@@ -58,7 +58,7 @@
             @test size(H) == (out_channel, N, batch_size)
             @test size(Y) == (pos_dim, N, batch_size)
 
-            g = Zygote.gradient(() -> sum(l(nf, ef)[1]), Flux.params(l))
+            g = gradient(() -> sum(l(nf, ef)[1]), Flux.params(l))
             @test length(g.grads) == 11
         end
 
@@ -70,7 +70,7 @@
             @test size(H) == (out_channel, N, batch_size)
             @test size(Y) == (pos_dim, N, batch_size)
 
-            g = Zygote.gradient(() -> sum(l(nf)[1]), Flux.params(l))
+            g = gradient(() -> sum(l(nf)[1]), Flux.params(l))
             @test length(g.grads) == 11
         end
     end

test/layers/positional.jl

@@ -23,7 +23,7 @@
             fg_ = l(fg)
             @test size(node_feature(fg_)) == (out_channel, N)
 
-            g = Zygote.gradient(() -> sum(node_feature(l(fg))), Flux.params(l))
+            g = gradient(() -> sum(node_feature(l(fg))), Flux.params(l))
             @test length(g.grads) == 4
         end
 
@@ -34,7 +34,7 @@
             Y = l(nf, ef)
             @test size(Y) == (out_channel, N, batch_size)
 
-            g = Zygote.gradient(() -> sum(l(nf, ef)), Flux.params(l))
+            g = gradient(() -> sum(l(nf, ef)), Flux.params(l))
             @test length(g.grads) == 2
         end
     end