From 4adef57b83440281a1326ca8c169a43e5f2e7c5a Mon Sep 17 00:00:00 2001 From: Aurora Rossi Date: Fri, 3 Jan 2025 14:22:36 +0100 Subject: [PATCH] Fix --- GNNLux/test/layers/pool.jl | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/GNNLux/test/layers/pool.jl b/GNNLux/test/layers/pool.jl index 69dee0926..48f9ee65c 100644 --- a/GNNLux/test/layers/pool.jl +++ b/GNNLux/test/layers/pool.jl @@ -8,13 +8,13 @@ x = randn(rng, Float32, in_dims, 10) @testset "GlobalPool" begin - l = GNNLux.GlobalPool(mean) + l = GlobalPool(mean) test_lux_layer(rng, l, g, x, sizey=(in_dims,1)) end @testset "GlobalAttentionPool" begin fgate = Dense(in_dims, 1) ffeat = Dense(in_dims, in_dims) - l = GNNLux.GlobalAttentionPool(fgate, ffeat) + l = GlobalAttentionPool(fgate, ffeat) test_lux_layer(rng, l, g, x, sizey=(in_dims,1), container=true) end end