def convert_dgl_to_nx(G: dgl.DGLGraph) -> nx.Graph: """ Converts a DGL Graph (``dgl.DGLGraph``) to a NetworkX (``nx.Graph``) object. Preserves node and edge attributes. :param G: ``dgl.DGLGraph`` to convert to ``NetworkX`` graph. :type G: dgl.DGLGraph :return: NetworkX graph object. :rtype: nx.Graph """ node_attrs = G.node_attr_schemes().keys() edge_attrs = G.edge_attr_schemes().keys() return dgl.to_networkx(G, node_attrs, edge_attrs)
def convert_dgl_to_nx(G: dgl.DGLGraph) -> nx.Graph: """ Converts a DGL Graph (dgl.DGLGraph) to a NetworkX (nx.Graph) object. Preservers node and edge attributes. :param G: dgl.DGLGraph to convert to NetworkX :type G: dgl.DGLGraph :return: NetworkX graph object :rtype: nx.Graph """ node_attrs = G.node_attr_schemes().keys() edge_attrs = G.edge_attr_schemes().keys() nx_g = dgl.to_networkx(G, node_attrs, edge_attrs) return nx_g