CI fails: TypeError: RLOOTrainer._get_per_token_logps_and_entropies() got an unexpected keyword argument 'token_type_ids'

[albertvillanova]

CI fails for all tests using tiny-Gemma3ForConditionalGeneration: https://github.com/huggingface/trl/actions/runs/18026398970/job/51294192894

TypeError: RLOOTrainer._get_per_token_logps_and_entropies() got an unexpected keyword argument 'token_type_ids'

FAILED tests/test_rloo_trainer.py::RLOOTrainerTester::test_training_vlm_0_trl_internal_testing_tiny_Gemma3ForConditionalGeneration - TypeError: _get_per_token_logps_and_entropies() got an unexpected keyword argument 'token_type_ids'
= 1 failed, 912 passed, 58 skipped, 3 xfailed, 142 warnings, 7 rerun in 815.53s (0:13:35) =