0a4e8b32b6b31f2b833476fe5ab129075ee632e2,python/dgl/heterograph.py,,find_src_dst_ntypes,#Any#Any#,4499

Before Change

        srctypes = {ntypes[tid] : tid for tid in src}
        dsttypes = {ntypes[tid] : tid for tid in dst}
        // handle isolated node types
        for ntid, ntype in enumerate(ntypes):
            if ntype not in srctypes and ntype not in dsttypes:
                srctypes[ntype] = ntid
        return srctypes, dsttypes
    else:
        return None

After Change

    if ret is None:
        return None
    else:
        src, dst = ret
        srctypes = {ntypes[tid.data] : tid.data for tid in src}
        dsttypes = {ntypes[tid.data] : tid.data for tid in dst}
        return srctypes, dsttypes

Instances