def load_proteins(dataset):
    """
    This function downloads, converts, and load Proteins graph dataset
    Parameter
    ---------
    dataset: output folder name
    """
    part_dir = dataset
    graph_file = os.path.join(part_dir + "/graph.dgl")

    if not os.path.exists("proteins.mtx"):
        download_proteins()
    if not os.path.exists(graph_file):
        g = proteins_mtx2dgl()
        save(g, dataset)
    ## load
    graph = load_graphs(graph_file)[0][0]
    return graph

    This function downloads, converts, and load Proteins graph dataset
    Parameter
    ---------
    dataset: output folder name
    
python/dgl/distgnn/tools/tools.py