fix gcn_con

JuliaGraphs · Jul 28, 2024 · e45fd98 · e45fd98
1 parent 2b3962a
commit e45fd98
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/GNNlib/src/layers/conv.jl b/GNNlib/src/layers/conv.jl
@@ -71,9 +71,9 @@ function gcn_conv(l, g::AbstractGNNGraph, x, edge_weight::EW, norm_fn::F, conv_w
 end
 
 # when we also have edge_weight we need to convert the graph to COO
-function gcn_conv(l, g::GNNGraph{<:ADJMAT_T}, x::AbstractMatrix, edge_weight::AbstractVector, norm_fn::F) where F    
+function gcn_conv(l, g::GNNGraph{<:ADJMAT_T}, x::AbstractMatrix, edge_weight::AbstractVector; kws...)    
     g = GNNGraph(edge_index(g)...; g.num_nodes)  # convert to COO
-    return gcn_conv(l, g, x, edge_weight, norm_fn)
+    return gcn_conv(l, g, x, edge_weight; kws...)
 end
 
 function cheb_conv(l, g::GNNGraph, X::AbstractMatrix{T}) where {T}