diff --git a/tests/test_online_dpo_trainer.py b/tests/test_online_dpo_trainer.py index 26ffc452930..72b68180df6 100644 --- a/tests/test_online_dpo_trainer.py +++ b/tests/test_online_dpo_trainer.py @@ -100,7 +100,7 @@ def tokenize(element): remove_unused_columns=False, gradient_accumulation_steps=1, learning_rate=9e-1, - evaluation_strategy="steps", + eval_strategy="steps", ) trainer = OnlineDPOTrainer(