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()

[((9, 4, 9, 10), 'gssl.utils.seed', 'seed', ({}, {}), '()', False, 'from gssl.utils import seed\n'), ((16, 8, 16, 18), 'gssl.inductive.datasets.load_ppi', 'load_ppi', ({}, {}), '()', False, 'from gssl.inductive.datasets import load_ppi\n'), ((18, 8, 18, 39), 'gssl.datasets.load_dataset', 'load_dataset', (), '', False, 'from gssl.datasets import load_dataset\n')]