fix tests

turbo-llm · Dec 13, 2024 · cc063a9 · cc063a9
1 parent 627b1f0
commit cc063a9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/turbo_alignment/trainers/dpo.py b/turbo_alignment/trainers/dpo.py
@@ -457,7 +457,7 @@ def compute_loss(
         return loss, chosen_rewards, rejected_rewards
 
 
-@DPOLossRegistry.register(DPOLossesType.NCA_PAIT)
+@DPOLossRegistry.register(DPOLossesType.NCA_PAIR)
 class NCAPairLoss(DPOLossRegistry):
     def __init__(self, *args, beta: float = 0.1, **kwargs) -> None:
         self.beta = beta