diff --git a/src/graph_helpers.py b/src/graph_helpers.py index 347c268d13a00ad3aac2317ab83a3c1cb9f1ef5f..07a652923bed36a80e48174a8c06bbf878670183 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