Unverified Commit 8c79885d authored by Da Zheng's avatar Da Zheng Committed by GitHub
Browse files

[BUGFIX] fix the error message in the data loader. (#509)

* fix.

* Change to required=True

* Update description

* Change back
parent 2ff8ecfb
...@@ -15,7 +15,7 @@ from .gindt import GINDataset ...@@ -15,7 +15,7 @@ from .gindt import GINDataset
def register_data_args(parser): def register_data_args(parser):
parser.add_argument("--dataset", type=str, required=False, parser.add_argument("--dataset", type=str, required=False,
help="The input dataset.") help="The input dataset. Can be cora, citeseer, pubmed, syn(synthetic dataset) or reddit")
citegrh.register_args(parser) citegrh.register_args(parser)
def load_data(args): def load_data(args):
...@@ -27,7 +27,7 @@ def load_data(args): ...@@ -27,7 +27,7 @@ def load_data(args):
return citegrh.load_pubmed() return citegrh.load_pubmed()
elif args.dataset == 'syn': elif args.dataset == 'syn':
return citegrh.load_synthetic(args) return citegrh.load_synthetic(args)
elif args.dataset.startswith('reddit'): elif args.dataset is not None and args.dataset.startswith('reddit'):
return RedditDataset(self_loop=('self-loop' in args.dataset)) return RedditDataset(self_loop=('self-loop' in args.dataset))
else: else:
raise ValueError('Unknown dataset: {}'.format(args.dataset)) raise ValueError('Unknown dataset: {}'.format(args.dataset))
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment