diff --git a/cogdl/wrappers/data_wrapper/pretraining/gcc_dw.py b/cogdl/wrappers/data_wrapper/pretraining/gcc_dw.py index a039560f..f014859d 100644 --- 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 index edb07f26..49e97d02 100644 --- 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