fix dpo eval override to call grandparent instead of the broken super (#2628) [skip ci]

This commit is contained in:
Wing Lian
2025-05-06 11:18:25 -04:00
committed by GitHub
parent 679743087a
commit ddaebf8309

View File

@@ -247,7 +247,9 @@ class AxolotlDPOTrainer(RngLoaderMixin, SchedulerMixin, DPOTrainer):
) )
# Base evaluation # Base evaluation
initial_output = super().evaluation_loop( initial_output = super( # pylint: disable=bad-super-call
DPOTrainer, self
).evaluation_loop(
dataloader, dataloader,
description, description,
prediction_loss_only, prediction_loss_only,