From cc063a954bb3805ba8dc06961e98a6c757f6bf62 Mon Sep 17 00:00:00 2001
From: Alexey Malakhov <131314005+alekseymalakhov11@users.noreply.github.com>
Date: Fri, 13 Dec 2024 17:35:58 +0300
Subject: [PATCH] fix tests

---
 turbo_alignment/trainers/dpo.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/turbo_alignment/trainers/dpo.py b/turbo_alignment/trainers/dpo.py
index 40e9091..2278eee 100755
--- 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