Function from_networkx

easygraph/convert.py:324–352

Convert a NetworkX graph to an EasyGraph graph. Args: g (Union[nx.Graph, nx.DiGraph]): A NetworkX graph Returns: Union[Graph, DiGraph]: Converted EasyGraph graph

(g: "Union[nx.Graph, nx.DiGraph]")

def from_networkx(g: "Union[nx.Graph, nx.DiGraph]") -> "Union[Graph, DiGraph]":
    """Convert a NetworkX graph to an EasyGraph graph.

    Args:
        g (Union[nx.Graph, nx.DiGraph]): A NetworkX graph

    Returns:
        Union[Graph, DiGraph]: Converted EasyGraph graph
    """
    # try:
    #     import networkx as nx
    # except ImportError:
    #     raise ImportError("NetworkX not found. Please install it.")
    if g.is_directed():
        G = eg.DiGraph()
    else:
        G = eg.Graph()

    # copy attributes
    G.graph = deepcopy(g.graph)

    nodes_with_edges = set()
    for v1, v2, _ in g.edges:
        G.add_edge(v1, v2)
        nodes_with_edges.add(v1)
        nodes_with_edges.add(v2)
    for node in set(g.nodes) - nodes_with_edges:
        G.add_node(node)
    return G


def to_dgl(g: "Union[Graph, DiGraph]"):