From 5c7406253b165c7c1efcd1d0b55e3f4b3c3f8871 Mon Sep 17 00:00:00 2001 From: v3nividiv1ci Date: Wed, 21 Jan 2026 11:26:26 +0000 Subject: [PATCH] fix: remove duplicate setup in gsm8k_rl --- examples/math/gsm8k_rl.py | 15 ++++----------- 1 file changed, 4 insertions(+), 11 deletions(-) diff --git a/examples/math/gsm8k_rl.py b/examples/math/gsm8k_rl.py index 5cfc9b3999..e095fd482b 100644 --- a/examples/math/gsm8k_rl.py +++ b/examples/math/gsm8k_rl.py @@ -10,18 +10,11 @@ def main(args): config, _ = load_expr_config(args, GRPOConfig) tokenizer = load_hf_tokenizer(config.tokenizer_path) - train_dataset = get_custom_dataset("train", config.train_dataset, tokenizer) - valid_dataset = get_custom_dataset("test", config.valid_dataset, tokenizer) - - workflow_kwargs = dict( - reward_fn="areal.reward.gsm8k.gsm8k_reward_fn", - gconfig=config.gconfig, - tokenizer=config.tokenizer_path, - enable_thinking=False, + train_dataset = get_custom_dataset( + split="train", + dataset_config=config.train_dataset, + tokenizer=tokenizer, ) - eval_workflow_kwargs = workflow_kwargs.copy() - eval_workflow_kwargs["gconfig"] = config.gconfig.new(temperature=0.6) - valid_dataset = get_custom_dataset( split="test", dataset_config=config.valid_dataset,