update doc

SuperXiang · SuperXiang · commit 4a7903c933c0 · 2021-05-13T20:09:27.000+08:00
diff --git a/docs/api_doc/featurizers.rst b/docs/api_doc/featurizers.rst
@@ -4,19 +4,6 @@ pahelix.featurizers
 
 .. contents:: Table of Contents
 
-het_gnn_featurizer
-==================
-
-.. automodule:: pahelix.featurizers.het_gnn_featurizer
-
-.. autoclass:: pahelix.featurizers.het_gnn_featurizer.DDiFeaturizer
-   :members:
-   :inherited-members:
-
-.. autofunction:: pahelix.featurizers.het_gnn_featurizer.num_nodes_stat
-
-.. autofunction:: pahelix.featurizers.het_gnn_featurizer.nx_graph_build
-
 pretrain_gnn_featurizer
 =======================
 
diff --git a/docs/api_doc/networks.rst b/docs/api_doc/networks.rst
@@ -9,10 +9,10 @@ basic_block
 
 .. automodule:: pahelix.networks.basic_block
 
-.. autoclass:: pahelix.networks.compound_encoder.Activation
+.. autoclass:: pahelix.networks.basic_block.Activation
    :members:
 
-.. autoclass:: pahelix.networks.compound_encoder.MLP
+.. autoclass:: pahelix.networks.basic_block.MLP
    :members:
 
 compound_encoder
diff --git a/docs/conf.py b/docs/conf.py
@@ -25,7 +25,7 @@
 language = 'en'
 
 # Import mock dependencies packages
-autodoc_mock_imports = ['paddle', 'pgl', 'rdkit', 'numpy', 'random', 'pandas', 'sklearn', 'networkx']
+autodoc_mock_imports = ['paddle', 'pgl', 'rdkit', 'numpy', 'random', 'pandas', 'sklearn', 'networkx', 'pdb', 'math', 'enum', 'collections', 'json', 'pickle', 'argparse', 're']
 
 
 # -- General configuration ---------------------------------------------------
diff --git a/docs/requirements.txt b/docs/requirements.txt
@@ -3,4 +3,7 @@ sphinxcontrib-httpdomain
 paddlepaddle>=2.0.0rc0
 pgl>=1.2.0
 numpy
-pandas
+pandas
+math
+random
+collections
