Spaces:
Runtime error
Runtime error
Update src/gen/gen_judgment.py
Browse files- src/gen/gen_judgment.py +1 -1
src/gen/gen_judgment.py
CHANGED
|
@@ -194,7 +194,7 @@ if __name__ == "__main__":
|
|
| 194 |
count += 1
|
| 195 |
continue
|
| 196 |
|
| 197 |
-
kwargs["answer"] = model_answers
|
| 198 |
if ref_answers:
|
| 199 |
kwargs["reference"] = [ref_answer[question_id] for ref_answer in ref_answers]
|
| 200 |
assert len(kwargs["reference"]) == len(configs["ref_model"])
|
|
|
|
| 194 |
count += 1
|
| 195 |
continue
|
| 196 |
|
| 197 |
+
kwargs["answer"] = model_answers.get(model,{question_id:''})[question_id]
|
| 198 |
if ref_answers:
|
| 199 |
kwargs["reference"] = [ref_answer[question_id] for ref_answer in ref_answers]
|
| 200 |
assert len(kwargs["reference"]) == len(configs["ref_model"])
|