Rodrigo Ferreira Rodrigues commited on
Commit
ca19a65
·
1 Parent(s): 3c35004

handling case when answer is not a list

Browse files
Files changed (1) hide show
  1. place_gen_evaluate.py +2 -0
place_gen_evaluate.py CHANGED
@@ -118,6 +118,8 @@ class Place_gen_evaluate(evaluate.Metric):
118
 
119
  try:
120
  f_ans = ast.literal_eval(gen)
 
 
121
  except:
122
  subs = ['[', ']', '(', ')', ' ']
123
  pattern = r'(' + '|'.join(map(re.escape, subs)) + r')'
 
118
 
119
  try:
120
  f_ans = ast.literal_eval(gen)
121
+ if not isinstance(f_ans, list):
122
+ f_ans = [f_ans]
123
  except:
124
  subs = ['[', ']', '(', ')', ' ']
125
  pattern = r'(' + '|'.join(map(re.escape, subs)) + r')'