From 843050c27775ac2ba62f6d5aad0c33386274ff1b Mon Sep 17 00:00:00 2001
From: Kamyar Salahi <kam.salahi@berkeley.edu>
Date: Mon, 21 Oct 2024 17:07:59 -0700
Subject: [PATCH] Fixing tests

---
 tests/test_supervised.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/test_supervised.py b/tests/test_supervised.py
index e1d9098d2..b8bec4f45 100644
--- a/tests/test_supervised.py
+++ b/tests/test_supervised.py
@@ -18,7 +18,7 @@ def test_supervised_eval():
     if tokenizer.pad_token is None:
         tokenizer.pad_token = tokenizer.eos_token
 
-    output = preprocess_supervised_example(examples, tokenizer)
+    output = preprocess_supervised_example(examples, tokenizer, "input", "output")
     assert len(output["input_ids"][0]) == output["sources_len"][0] + 1
 
     ex = {