From 030366e162b94ce5eaff721bf9768cd6c69d7d6f Mon Sep 17 00:00:00 2001 From: Yukuo Cen Date: Thu, 28 Oct 2021 14:36:47 +0800 Subject: [PATCH] Fix building docs --- cogdl/wrappers/data_wrapper/pretraining/gcc_dw.py | 3 +-- docs/requirements.txt | 8 ++++---- 2 files changed, 5 insertions(+), 6 deletions(-) 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