fix doctests

Author: CarloLucibello

GNNLux/docs/make.jl

@@ -12,6 +12,10 @@ using GNNLux
 using Lux, GNNGraphs, GNNlib, Graphs
 using DocumenterInterLinks
 
+ENV["DATADEPS_ALWAYS_ACCEPT"] = true # for MLDatasets
+
+DocMeta.setdocmeta!(GNNGraphs, :DocTestSetup, :(using GNNGraphs, MLUtils); recursive = true)
+DocMeta.setdocmeta!(GNNlib, :DocTestSetup, :(using GNNlib); recursive = true)
 DocMeta.setdocmeta!(GNNLux, :DocTestSetup, :(using GNNLux); recursive = true)
 
 mathengine = MathJax3(Dict(:loader => Dict("load" => ["[tex]/require", "[tex]/mathtools"]),

GNNLux/docs/src/api/pool.md

@@ -5,13 +5,6 @@ CollapsedDocStrings = true
 
 # Pooling Layers
 
-## Index
-
-```@index
-Order = [:type, :function]
-Pages   = ["pool.md"]
-```
-
 ```@autodocs
 Modules = [GNNLux]
 Pages   = ["layers/pool.jl"]

GNNLux/src/layers/basic.jl

@@ -31,9 +31,14 @@ julia> using Lux, GNNLux, Random
 
 julia> rng = Random.default_rng();
 
-julia> m = GNNChain(GCNConv(2=>5), 
-                    x -> relu.(x), 
-                    Dense(5=>4))
+julia> m = GNNChain(GCNConv(2 => 5, relu), Dense(5 => 4))
+GNNChain(
+    layers = NamedTuple(
+        layer_1 = GCNConv(2 => 5, relu),  # 15 parameters
+        layer_2 = Dense(5 => 4),        # 24 parameters
+    ),
+)         # Total: 39 parameters,
+          #        plus 0 states.
 
 julia> x = randn(rng, Float32, 2, 3);
 
@@ -44,8 +49,10 @@ GNNGraph:
 
 julia> ps, st = LuxCore.setup(rng, m);
 
-julia> m(g, x, ps, st)     # First entry is the output, second entry is the state of the model
-(Float32[-0.15594329 -0.15594329 -0.15594329; 0.93431795 0.93431795 0.93431795; 0.27568763 0.27568763 0.27568763; 0.12568939 0.12568939 0.12568939], (layer_1 = NamedTuple(), layer_2 = NamedTuple(), layer_3 = NamedTuple()))
+julia> y, st = m(g, x, ps, st);     # First entry is the output, second entry is the state of the model
+
+julia> size(y)
+(4, 3)
 ```
 """
 @concrete struct GNNChain <: GNNContainerLayer{(:layers,)}

GNNlib/docs/make.jl

@@ -12,6 +12,10 @@ using GNNGraphs
 import Graphs
 using DocumenterInterLinks
 
+ENV["DATADEPS_ALWAYS_ACCEPT"] = true # for MLDatasets
+DocMeta.setdocmeta!(GNNGraphs, :DocTestSetup, :(using GNNGraphs, MLUtils); recursive = true)
+DocMeta.setdocmeta!(GNNlib, :DocTestSetup, :(using GNNlib); recursive = true)
+
 assets=[]
 prettyurls = get(ENV, "CI", nothing) == "true"
 mathengine = MathJax3()

GraphNeuralNetworks/docs/src/api/pool.md

@@ -5,13 +5,6 @@ CollapsedDocStrings = true
 
 # Pooling Layers
 
-## Index
-
-```@index
-Order = [:type, :function]
-Pages   = ["pool.md"]
-```
-
 ```@autodocs
 Modules = [GraphNeuralNetworks]
 Pages   = ["layers/pool.jl"]