Follow the naming convention for dataset name.

Adamliu1 · Jul 1, 2024 · c388fb6 · c388fb6
1 parent bfc5063
commit c388fb6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/llm_unlearn_ucl/unlearn_harm.py b/llm_unlearn_ucl/unlearn_harm.py
@@ -400,7 +400,7 @@ def main(args) -> None:
 
         question_prefix_str = "### Question:"
         answer_prefix_str = "### Answer:"
-    elif args.unlearning_dataset == "math_qa":
+    elif args.unlearning_dataset == "allenai/math_qa":
         full_bad_dataset = load_dataset("math_qa", split="train")
         if args.shuffle_seed:
             # shuffle the dataset with a given seed for reproducibility