mirror of
https://github.com/NousResearch/atropos.git
synced 2026-04-19 12:57:58 +00:00
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
This commit is contained in:
parent
e28297b625
commit
9d3dbd1a73
1 changed files with 3 additions and 1 deletions
|
|
@ -197,7 +197,9 @@ class MCQAThinkingEnv(BaseEnv):
|
|||
{
|
||||
"messages": tuple(trajectory_messages),
|
||||
"expected_answer": item[1], # Letter (A, B, C, D)
|
||||
"ground_truth_text": item[2], # Include the answer_string/ground_truth_text
|
||||
"ground_truth_text": item[
|
||||
2
|
||||
], # Include the answer_string/ground_truth_text
|
||||
"finish_reason": completion_choice.finish_reason, # Add the stop reason
|
||||
"tokens": nodes[i].tokens,
|
||||
"masks": nodes[i].masked_tokens,
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue