Merge branch 'ar/add-pool-layers' of https://github.com/JuliaGraphs/G…

…raphNeuralNetworks.jl into ar/add-pool-layers
JuliaGraphs · Jan 6, 2025 · b8c479d · b8c479d
2 parents c048532 + e80872c
commit b8c479d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/GNNLux/src/layers/pool.jl b/GNNLux/src/layers/pool.jl
@@ -130,4 +130,4 @@ function TopKPool(adj::AbstractMatrix, k::Int, in_channel::Int; init = glorot_un
     TopKPool(adj, k, init(in_channel), similar(adj, k, k))
 end
 
-(t::TopKPool)(x::AbstractArray, ps, st) = GNNlib.topk_pool(t, x)
+(t::TopKPool)(x::AbstractArray, ps, st) = GNNlib.topk_pool(t, x), st