diff --git a/docs/source/api/python/data.rst b/docs/source/api/python/data.rst index 1942ce901e59..f9a371badad9 100644 --- a/docs/source/api/python/data.rst +++ b/docs/source/api/python/data.rst @@ -33,6 +33,14 @@ Mini graph classification dataset .. autoclass:: MiniGCDataset :members: __getitem__, __len__, num_classes +Graph kernel dataset +```````````````````` + +For more information about the dataset, see `Benchmark Data Sets for Graph Kernels `__. + +.. autoclass:: TUDataset + :members: __getitem__, __len__ + Protein-Protein Interaction dataset ``````````````````````````````````` diff --git a/docs/source/conf.py b/docs/source/conf.py index 6f7cefa9fd91..6cb5884e4d4f 100644 --- a/docs/source/conf.py +++ b/docs/source/conf.py @@ -215,7 +215,8 @@ } # Compatibility for different backend when builds tutorials -if os.environ['DGLBACKEND'] == 'mxnet': +dglbackend = os.environ.get("DGLBACKEND", "") +if dglbackend == 'mxnet': sphinx_gallery_conf['filename_pattern'] = "/*(?<=mx)\.py" -if os.environ['DGLBACKEND'] == 'pytorch': +if dglbackend == 'pytorch': sphinx_gallery_conf['filename_pattern'] = "/*(?