Documentation examples incorrect about geometricflux.jl HOT 1 OPEN

model = Chain(
    GCNConv(1024=>256, relu),
    x -> (node_feature(x), global_feature(x)),
    (nf, gf) -> (softmax(nf), identity.(gf))
) # this works

model(some_featuredgraph) # ❌ this should work, but does not

get_nf_gf(fg) = (node_feature(fg), global_feature(fg))

model = Chain(
    GCNConv(1024=>256, relu),
    get_nf_gf,
    (nf, gf) -> (softmax(nf), identity.(gf))
)

model(some_featuredgraph) # ✔️