Skip to content

Commit f9e2ea3

Browse files
SurbhiJainUSCA9isha
authored andcommitted
Update condition for dataset split
1 parent f76ae69 commit f9e2ea3

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

src/maxtext/trainers/post_train/rl/train_rl.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -337,7 +337,7 @@ def prepare_datasets(trainer_config, model_tokenizer):
337337
"nvidia/OpenMathReasoning",
338338
"open-r1/OpenR1-Math-220k",
339339
"bethgelab/CuratedThoughts",
340-
] and (not eval_dataset_name or eval_dataset_name == trainer_config.dataset_name):
340+
] and eval_dataset_name == trainer_config.dataset_name:
341341
import datasets # pylint: disable=import-outside-toplevel
342342

343343
splits = prepare_train_and_eval_dataset(trainer_config)

0 commit comments

Comments
 (0)