Fixing tests

stanford-crfm · Oct 22, 2024 · 843050c · 843050c
1 parent b7bbc40
commit 843050c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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 = {