From 7acb5ca8156b74ba7b68a856de75d7136dd6b755 Mon Sep 17 00:00:00 2001
From: Austin Liu <austin362667@gmail.com>
Date: Thu, 5 Dec 2024 23:08:00 +0800
Subject: [PATCH] Fix `distillation_loss` arg typo

Signed-off-by: Austin Liu <austin362667@gmail.com>
---
 test/utils.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/test/utils.py b/test/utils.py
index fc5ba4944..69f8a8c40 100644
--- a/test/utils.py
+++ b/test/utils.py
@@ -536,7 +536,7 @@ def __init__(
         self.temperature = temperature
 
     @abstractmethod
-    def distillation_loss(self, student_logps, teacher_logps):
+    def distillation_loss(self, student_logits, teacher_logits):
         """Abstract method for computing distillation loss."""
         pass