Fix

JuliaGraphs · Jan 3, 2025 · 4adef57 · 4adef57
1 parent a173306
commit 4adef57
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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