fix: wrong split for truthful_qa

Adamliu1 · Jul 10, 2024 · 958c396 · 958c396
1 parent 0d7e803
commit 958c396
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/llm_unlearn_ucl/data_utils.py b/llm_unlearn_ucl/data_utils.py
@@ -26,7 +26,7 @@ def make_dataset(
     )
     if dataset_uri == "truthfulqa/truthful_qa":
         full_dataset = cast(
-            Dataset, load_dataset(dataset_uri, "generation", split=split)
+            Dataset, load_dataset(dataset_uri, "generation", split="validation")
         )
     full_dataset = cast(
         Dataset, load_dataset(dataset_uri, split=split)