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

Before Change

        a dictionary from type name to type id. Return None if the graph is
        not uni-bipartite.
    
    src, dst, _ = metagraph.edges()
    if set(src.tonumpy()).isdisjoint(set(dst.tonumpy())):
        srctypes = {ntypes[tid] : tid for tid in src}
        dsttypes = {ntypes[tid] : tid for tid in dst}
        // handle isolated node types

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