18a26fcfb1983af7fba69db9bdce7ba5e6a9945f,python/dgl/data/graph_serialize.py,,load_labels,#Any#,149

Before Change

    >>> label_dict = load_graphs("./data.bin")

    
    metadata = _CAPI_DGLLoadGraphs(filename, [], True)
    label_dict = {}
    for k, v in metadata.labels.items():
        label_dict[k] = F.zerocopy_from_dgl_ndarray(v.data)
    return label_dict

After Change

    
    version = _CAPI_GetFileVersion(filename)
    if version == 1:
        return load_labels_v1(filename)
    elif version == 2:
        return load_labels_v2(filename)
    else:
        raise Exception("Invalid DGL Version Number")


def load_labels_v2(filename):
    Internal functions for loading labels from V2 format
    label_dict = {}
    nd_dict = _CAPI_LoadLabels_V2(filename)

Instances