diff --git a/src/abstinent.py b/src/abstinent.py index df2696519f2b30ccc21562fcc247ec4368d21af4..29201f1608cfd9b57e04910ea53d395666111924 100644 --- a/src/abstinent.py +++ b/src/abstinent.py @@ -373,8 +373,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+')')