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
b9dac318ee
commit
3cb09259df
1 changed files with 1 additions and 3 deletions
|
|
@ -666,9 +666,7 @@ class CodingEnv(BaseEnv):
|
|||
async def setup(self):
|
||||
"""Setup the environment"""
|
||||
if self.config.dataset_name == "deepmind":
|
||||
self.train = load_dataset(
|
||||
"deepmind/code_contests", split="train"
|
||||
) # CHANGE
|
||||
self.train = load_dataset("deepmind/code_contests", split="train") # CHANGE
|
||||
else:
|
||||
self.train = load_dataset(
|
||||
"NousResearch/RLVR_Coding_Problems", split="train"
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue