18a26fcfb1983af7fba69db9bdce7ba5e6a9945f,python/dgl/data/graph_serialize.py,,save_graphs,#Any#Any#Any#,63

Before Change

    
    if isinstance(g_list, DGLGraph):
        g_list = [g_list]
    if (labels is not None) and (len(labels) != 0):
        label_dict = dict()
        for key, value in labels.items():
            label_dict[key] = F.zerocopy_to_dgl_ndarray(value)

After Change

    
    g_sample = g_list[0] if isinstance(g_list, list) else g_list
    if isinstance(g_sample, DGLGraph):
        save_dglgraphs(filename, g_list, labels)
    elif isinstance(g_sample, DGLHeteroGraph):
        save_heterographs(filename, g_list, labels)
    else:
        raise Exception(
            "Invalid argument g_list. Must be a DGLGraph or a list of DGLGraphs/DGLHeteroGraphs")


def save_dglgraphs(filename, g_list, labels=None):
    Internal function to save DGLGraphs
    if isinstance(g_list, DGLGraph):
        g_list = [g_list]

Instances