Skip to content

Commit 1cbfb00

Browse files
Replace remaining trainer.tokenizer with trainer.processing_class in GRPO test (#4192)
1 parent e086f07 commit 1cbfb00

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/test_grpo_trainer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1882,7 +1882,7 @@ def test_update_with_inputs_different_seq_len(self):
18821882
Test with inputs where the sequence lengths are different from the prepopulated buffer.
18831883
"""
18841884
self._prepopulate_buffer()
1885-
pad_token_id = self.trainer.tokenizer.pad_token_id
1885+
pad_token_id = self.trainer.processing_class.pad_token_id
18861886
group_advantages = torch.tensor([[0.6, 0.6], [0.3, 0.45]]) # one no-variance, one variance
18871887
inputs = {
18881888
"group_advantages": group_advantages,

0 commit comments

Comments
 (0)