NousResearch/atropos
1
0
Fork 0
mirror of https://github.com/NousResearch/atropos.git synced 2026-05-01 17:45:16 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

[pre-commit.ci] auto fixes from pre-commit.com hooks

Browse source 
for more information, see https://pre-commit.ci
This commit is contained in:
pre-commit-ci[bot] 2025-11-14 06:14:18 +00:00
parent be74c759e5
commit 0a3c15c7ad
1 changed files with 4 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

6
environments/swe_rl_env.py
View file

@ -861,7 +861,7 @@ class SWERLEnv(BaseEnv):
tokens = rollout_item["tokens"]
masks = rollout_item["masks"]
logprobs = rollout_item["logprobs"]
assistant_response = ""
if (
trajectory_messages
@ -1167,7 +1167,9 @@ class SWERLEnv(BaseEnv):
score_for_rollout = scored_data["scores"][i]
failed_rollouts_with_scores_to_save.append(
{
"conversation": rollout_item["messages"], # Full conversation history
"conversation": rollout_item[
"messages"
], # Full conversation history
"score": score_for_rollout,
"oracle_patch": rollout_item["oracle_patch"],
"finish_reason": rollout_item["finish_reason"],