NousResearch/atropos
1
0
Fork 0
mirror of https://github.com/NousResearch/atropos.git synced 2026-04-19 12:57:58 +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-13 08:04:36 +00:00
parent 3f6265563f
commit b03b8d3808
1 changed files with 1 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

3
environments/gsm8k_server.py
View file

@ -309,7 +309,6 @@ class GSM8kEnv(BaseEnv):
masks = item["masks"]
logprobs = item["logprobs"]
# remove obviously bad examples
if len([1 for i in masks if i != -100]) < 10:
continue
@ -317,7 +316,7 @@ class GSM8kEnv(BaseEnv):
scores["masks"].append(masks)
scores["inference_logprobs"].append(logprobs)
scores["scores"].append(1.0 if reward else -1.0)
if len(scores["tokens"]) >= self.config.group_size:
break