Fix TopK pool test

JuliaGraphs · Jan 7, 2025 · e575066 · e575066
1 parent b8c479d
commit e575066
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/GNNLux/test/layers/pool.jl b/GNNLux/test/layers/pool.jl
@@ -38,7 +38,7 @@
                 @test size(p.p) == (in_channel,)
                 @test eltype(p.Ã) === T
                 @test size(p.Ã) == (k, k)
-                y = p(X, ps, st)
+                y, st = p(X, ps, st)
                 @test size(y) == (in_channel, k)
             end
         end