Commit 6c361958 authored by Victor Zimmermann's avatar Victor Zimmermann
Browse files

Bugfix.

parent 5e6a4953

src/absinth.py

+2 −2
Original line number Diff line number Diff line
@@ -163,9 +163,9 @@ def induce(topic_name: str, result_list: list) -> (nx.Graph, list, dict): 
            new_dict = dict()

            for key,value in edge_freq_dict.items():

                if key[0] in new_dict:

                    key[0].add({key[1]:value})

                    new_dict[0].add({key[1]:value})

                else:

                    key[0] = {key[1]:value}

                    new_dict[0] = {key[1]:value}



            edge_freq_dict = new_dict

            json.dump(edge_freq_dict, edge_file)

src/abstinent.py

+2 −2
Original line number Diff line number Diff line
@@ -381,9 +381,9 @@ def induce(topic_name: str, result_list: list) -> (nx.Graph, list, dict): 
            new_dict = dict()

            for key,value in edge_freq_dict.items():

                if key[0] in new_dict:

                    key[0].add({key[1]:value})

                    new_dict[0].add({key[1]:value})

                else:

                    key[0] = {key[1]:value}

                    new_dict[0] = {key[1]:value}



            edge_freq_dict = new_dict

            json.dump(edge_freq_dict, edge_file)