Spaces:
Sleeping
Sleeping
Rodrigo Ferreira Rodrigues
commited on
Commit
·
ca19a65
1
Parent(s):
3c35004
handling case when answer is not a list
Browse files- 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')'
|