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

pipelineRL

Browse source
This commit is contained in:
Jai Suphavadeeprasit 2026-01-21 17:02:01 -05:00
parent e891a7f808
commit 806888d9d3
2 changed files with 9 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

4
environments/eval_environments/gsm8k_eval.py
View file

@ -133,10 +133,10 @@ class GSM8KEvalEnv(BaseEnv):
self,
config: GSM8KEvalConfig,
server_configs: List[APIServerConfig],
slurm_job_id: Optional[str] = None,
slurm=False,
testing: bool = False,
):
super().__init__(config, server_configs, slurm_job_id, testing)
super().__init__(config, server_configs, slurm, testing)
self.config: GSM8KEvalConfig = config
self.eval_items: List[Dict] = []
self._dataset_loaded = False