From 6c361958930bb684630274dbe20a5e245ec4782d Mon Sep 17 00:00:00 2001
From: zimmermann <zimmermann@cl.uni-heidelberg.de>
Date: Thu, 22 Mar 2018 18:57:21 +0100
Subject: [PATCH] Bugfix.
---
src/absinth.py | 4 ++--
src/abstinent.py | 6 +++---
2 files changed, 5 insertions(+), 5 deletions(-)
diff --git a/src/absinth.py b/src/absinth.py
index fb40141..0f93120 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 901b99e..99e4468 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)
--
GitLab