From 9f47bb77042fa7c4ca90b9cd6bb891bd2d45f440 Mon Sep 17 00:00:00 2001
From: opi <opitz@cl.uni-heidelberg.de>
Date: Tue, 23 Jul 2019 16:13:59 +0200
Subject: [PATCH] fixed the case where a trial has no category
---
src/graph_helpers.py | 6 +++++-
1 file changed, 5 insertions(+), 1 deletion(-)
diff --git a/src/graph_helpers.py b/src/graph_helpers.py
index 347c268..07a6529 100644
--- a/src/graph_helpers.py
+++ b/src/graph_helpers.py
@@ -41,7 +41,11 @@ def simplify_text_description_nodes(G,node_index_dict,mode="None",min_freq=1):
# we iterate over all trials
for i,tn in enumerate(trialnodes):
#get corresponding cat node
- catn = [n for n in G.neighbors(tn[0]) if isinstance(G.nodes[n]["nodeobj"],dh.OffenceNode)][0]
+ catn = [n for n in G.neighbors(tn[0]) if isinstance(G.nodes[n]["nodeobj"],dh.OffenceNode)]
+ if not catn:
+ continue
+ else:
+ catn=catn[0]
catn=[catn,G.nodes[catn]]
category=catn[1]["nodeobj"].d["category"]
descr_vectors = None
--
GitLab