fixes

Author: CarloLucibello

docs/make.jl

@@ -1,5 +1,9 @@
-using Flux, NNlib, GraphNeuralNetworks, Graphs, SparseArrays
+using GraphNeuralNetworks
 using GNNGraphs
+using Flux 
+using NNlib
+using Graphs
+using SparseArrays
 using Pluto, PlutoStaticHTML # for tutorials
 using Documenter, DemoCards
 using DocumenterInterLinks

docs/src/api/gnngraph.md

@@ -1,5 +1,5 @@
 ```@meta
-CurrentModule = GraphNeuralNetworks
+CurrentModule = GNNGraphs
 ```
 
 # GNNGraph
@@ -41,8 +41,7 @@ Private = false
 ```
 
 ```@docs
-Graphs.neighbors
-Graphs.has_edge
+Graphs.neighbors(::GNNGraph, ::Integer)
 ```
 
 ## Transform
@@ -79,7 +78,7 @@ Private = false
 ```
 
 ```@docs
-Graphs.intersect
+Base.intersect
 ```
 
 ## Sampling 

docs/src/api/heterograph.md

@@ -11,6 +11,10 @@ Pages   = ["gnnheterograph.jl"]
 Private = false
 ```
 
+```@docs
+Graphs.has_edge(::GNNHeteroGraph, ::Tuple{Symbol, Symbol, Symbol}, ::Integer, ::Integer)
+```
+
 ## Heterogeneous Graph Convolutions
 
 Heterogeneous graph convolutions are implemented in the type [`HeteroGraphConv`](@ref).