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

for more information, see https://pre-commit.ci
This commit is contained in:
pre-commit-ci[bot] 2025-12-31 21:05:41 +00:00
parent b9dac318ee
commit 3cb09259df

View file

@ -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"