apsys commited on
Commit
4f5b90a
·
verified ·
1 Parent(s): 2589956

Update src/gen/gen_judgment.py

Browse files
Files changed (1) hide show
  1. 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[model][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"])
 
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"])