[BUG] pass tests

tatsu-lab · Aug 26, 2024 · 3edfb28 · 3edfb28
1 parent 56a1c74
commit 3edfb28
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/test_pairwise_evaluator.py b/tests/test_pairwise_evaluator.py
@@ -74,6 +74,8 @@ def test_single_annotator(single_annotator, df_to_annotate):
         "output_2",
         "preference",
         "completions",
+        "preference_date",
+        "preference_version",
     }
     # check that you also save the completions.
     assert df_annotated["completions"].tolist() == parsable_completions