Commit 5b6b07d7 authored by Victor Zimmermann's avatar Victor Zimmermann
Browse files

Update absinth.py

parent 4135f407

src/absinth.py

+3 −3
Original line number Diff line number Diff line
@@ -725,17 +725,17 @@ def draw_propagation(graph: nx.Graph, root_hub_list: list) -> None: 


    pos = nx.spring_layout(graph, scale=3)

    

    nodes = list

    nodes = set()

    for root1 in root_hub_list:

        for root2 in root_hub_list:

            path = nx.shortest_path(graph, root1, root2)

            for node in path:

                nodes.append(node)

                nodes.add(node)

                

    edges = [edge for edge in graph.edges if edge[0] in nodes and edge[1] in nodes]

    



    mapping = {node:graph.node[node]['sense'] for node in set(nodes)}

    mapping = {node:graph.node[node]['sense'] for node in nodes}



    nx.draw_networkx_edges(graph, pos,

                           alpha=0.5,