import sys

from gssl.datasets import load_dataset
from gssl.inductive.datasets import load_ppi
from gssl.utils import seed


def main():
    seed()

    # Read dataset name
    dataset_name = sys.argv[1]

    # Load dataset
    if dataset_name == "PPI":
        load_ppi()
    else:
        load_dataset(name=dataset_name)


if __name__ == "__main__":
    main()

[((142, 148), 'gssl.utils.seed', 'seed', ([], {}), '()\n', (146, 148), False, 'from gssl.utils import seed\n'), ((263, 273), 'gssl.inductive.datasets.load_ppi', 'load_ppi', ([], {}), '()\n', (271, 273), False, 'from gssl.inductive.datasets import load_ppi\n'), ((292, 323), 'gssl.datasets.load_dataset', 'load_dataset', ([], {'name': 'dataset_name'}), '(name=dataset_name)\n', (304, 323), False, 'from gssl.datasets import load_dataset\n')]