Fix

Author: aurorarossi

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