@dataloader('PubMed')
def _load_pubmed():
    return tg.datasets.Planetoid(name='PubMed', root='/tmp/pubmed')[0]
@dataloader('PubMed')
def _load_pubmed():
    return tg.datasets.Planetoid(name='PubMed', root='/tmp/pubmed')[0]