Fix TopK pool test

aurorarossi · aurorarossi · commit e5750669c327 · 2025-01-07T11:22:03.000+01:00
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
