diff --git a/src/absinth.py b/src/absinth.py index cc11a35a2757e526cdaf887b7ac0edb2e0ddbb12..02de6f274986ea691d29a0f30ee8d22310415c9d 100644 --- a/src/absinth.py +++ b/src/absinth.py @@ -155,8 +155,9 @@ def induce(topic_name: str, result_list: list) -> (nx.Graph, list, dict): node_freq_dict, edge_freq_dict = frequencies(target_string, result_list) with open(config.graph+node_dict_name, 'w') as node_file, open(config.graph+edge_dict_name, 'w') as edge_file: - node_file.write(json.dumps(node_freq_dict)) - edge_file.write(json.dumps(edge_freq_dict)) + + json.dump(node_freq_dict, node_file) + json.dump(edge_freq_dict, edge_file) #builds graph from these dictionaries, also applies multiple filters print('[a]', 'Building graph.\t('+topic_name+')')