diff --git a/src/axolotl/core/trainers/dpo/trainer.py b/src/axolotl/core/trainers/dpo/trainer.py index 20bcc8dd6..1ce7deea7 100644 --- a/src/axolotl/core/trainers/dpo/trainer.py +++ b/src/axolotl/core/trainers/dpo/trainer.py @@ -247,7 +247,9 @@ class AxolotlDPOTrainer(RngLoaderMixin, SchedulerMixin, DPOTrainer): ) # Base evaluation - initial_output = super().evaluation_loop( + initial_output = super( # pylint: disable=bad-super-call + DPOTrainer, self + ).evaluation_loop( dataloader, description, prediction_loss_only,