diff --git a/src/absinth.py b/src/absinth.py index fb401415ae9cfc362830b8409e44d82f208761fb..0f93120be64d4e58fb652db0e09d11c4a14f41c4 100644 --- a/src/absinth.py +++ b/src/absinth.py @@ -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) diff --git a/src/abstinent.py b/src/abstinent.py index 901b99e8ad10aad372371b2f0fe0144f64e439b9..99e4468fdc311a2f05a2d22d7de81ef5c201a7e6 100644 --- a/src/abstinent.py +++ b/src/abstinent.py @@ -377,13 +377,13 @@ def induce(topic_name: str, result_list: list) -> (nx.Graph, list, dict): with open(config.graph+node_dict_name, 'w') as node_file, open(config.graph+edge_dict_name, 'w') as edge_file: json.dump(node_freq_dict, node_file) - + 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)