mirror of
https://github.com/NousResearch/atropos.git
synced 2026-04-19 12:57:58 +00:00
parent
cdc23ba5dc
commit
632ab0161c
22 changed files with 85 additions and 0 deletions
|
|
@ -383,6 +383,10 @@ class MathEnv(BaseEnv):
|
|||
to_postprocess = await self.score(to_score)
|
||||
if to_postprocess is None:
|
||||
return None, to_backlog
|
||||
if all(
|
||||
[to_postprocess["scores"][0] == score for score in to_postprocess["scores"]]
|
||||
):
|
||||
return None, to_backlog
|
||||
self.normal_rollouts.append(
|
||||
(
|
||||
prompt_format.format(prompt=problem_format.format(problem=item[0])),
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue