dmlc · mufeili · Oct 27, 2022 · Aug 16, 2022 · Aug 16, 2022 · Aug 19, 2022
diff --git a/docs/source/api/python/nn-pytorch.rst b/docs/source/api/python/nn-pytorch.rst
@@ -112,6 +112,7 @@ Utility Modules
     ~dgl.nn.pytorch.utils.JumpingKnowledge
     ~dgl.nn.pytorch.sparse_emb.NodeEmbedding
     ~dgl.nn.pytorch.explain.GNNExplainer
+    ~dgl.nn.pytorch.explain.HeteroGNNExplainer
     ~dgl.nn.pytorch.utils.LabelPropagation
 
 Network Embedding Modules

diff --git a/python/dgl/nn/pytorch/explain/__init__.py b/python/dgl/nn/pytorch/explain/__init__.py
@@ -1,6 +1,4 @@
 """Torch modules for explanation models."""
 # pylint: disable= no-member, arguments-differ, invalid-name
 
-from .gnnexplainer import GNNExplainer
-
-__all__ = ['GNNExplainer']
+from .gnnexplainer import GNNExplainer, HeteroGNNExplainer