Fix import name

for-ai · Jul 29, 2024 · d427903 · d427903
1 parent 56d74ca
commit d427903
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/scripts/utils.py b/scripts/utils.py
@@ -3,7 +3,7 @@
 
 import fastchat
 import transformers
-from dataset import Dataset, load_dataset
+from datasets import Dataset, load_dataset
 
 from rewardbench.utils import check_tokenizer_chat_template, prepare_dialogue
 from rewardbench.utils import prepare_dialogue_from_tokenizer