[Docs] Fix building docs (#293)

THUDM · Oct 28, 2021 · 0a64b59 · 0a64b59
1 parent e0e8f3d
commit 0a64b59
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 6 deletions.
diff --git a/cogdl/wrappers/data_wrapper/pretraining/gcc_dw.py b/cogdl/wrappers/data_wrapper/pretraining/gcc_dw.py
@@ -12,11 +12,10 @@
 
 from torch.utils.data import DataLoader
 
-from .. import register_data_wrapper, DataWrapper
+from .. import DataWrapper
 from cogdl.data import batch_graphs, Graph
 
 
-@register_data_wrapper("gcc_dw")
 class GCCDataWrapper(DataWrapper):
     @staticmethod
     def add_args(parser):

diff --git a/docs/requirements.txt b/docs/requirements.txt
@@ -1,7 +1,7 @@
-sphinx
-sphinx_rtd_theme
-sphinx-markdown-tables
-recommonmark
+sphinx==4.2.0
+sphinx_rtd_theme==1.0.0
+sphinx-markdown-tables==0.0.15
+recommonmark==0.7.1
 networkx
 matplotlib
 tqdm