Add export tokpool

aurorarossi · aurorarossi · commit c4986f653c74 · 2025-01-03T18:20:52.000+01:00
diff --git a/GNNLux/src/GNNLux.jl b/GNNLux/src/GNNLux.jl
@@ -50,7 +50,8 @@ export TGCN,
 
 include("layers/pool.jl")
 export GlobalPool,
-       GlobalAttentionPool
+       GlobalAttentionPool,
+       TopKPool
 
 end #module
  
