ImportError when running dev/grpo_full_finetune_distributed: batched_rewards function was removed

[HayrapetyanZhirayr]

Running the following command fails with an ImportError:

tune run --nproc_per_node 8 dev/grpo_full_finetune_distributed --config ./recipes/configs/dev/qwen3B_sync_grpo.yaml

The error occurs because the script grpo_full_finetune_distributed.py imports batched_rewards func, which was removed on main.

As a result, the script is currently broken.

[krammnic]

Hey! Thanks for the issue, I will try to repro and then write you back.

[hammoudhasan]

Same issue !

[krammnic]

@pytorchbot label bug

[krammnic]

I will open PR

[hammoudhasan]

@krammnic I managed to run the code by checking out to revert-2623-gh/IvanKobzarev/1/head branch. It might serve as a good reference in that case.

[pytorch-bot]

Didn't find following labels among repository labels: critical

[krammnic]

@pytorchbot label high-priority