wzkariampuzha commited on
Commit
acee9d8
1 Parent(s): 38c6f3a

Update extract_abs.py

Browse files
Files changed (1) hide show
  1. extract_abs.py +1 -1
extract_abs.py CHANGED
@@ -156,7 +156,7 @@ def parse_info(sentences:List[str], model_outputs:List[List[Union[Dict[str,str],
156
  for entity, output in output_dict.items():
157
  if not output:
158
  output_dict[entity] = None
159
- else:
160
  #remove duplicates from list but keep ordering instead of using sets
161
  output = list(OrderedDict.fromkeys(output))
162
  output_dict[entity] = output
 
156
  for entity, output in output_dict.items():
157
  if not output:
158
  output_dict[entity] = None
159
+ elif entity !='STAT':
160
  #remove duplicates from list but keep ordering instead of using sets
161
  output = list(OrderedDict.fromkeys(output))
162
  output_dict[entity] = output