Merge pull request #86 from HallerPatrick/main

Fix loading of dataset subset
flairNLP · Dec 23, 2023 · 8dc2332 · 8dc2332
2 parents 56c9cb9 + 3ab3b18
commit 8dc2332
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_prompts.py b/tests/test_prompts.py
@@ -126,7 +126,7 @@ def preprocess_qa(example):
         self.question_answering = load_dataset("squad", split="train").flatten()\
             .rename_column("answers.text", "answer").map(preprocess_qa).select([1, 2, 3])
         self.ner = load_dataset("conll2003", split="train").select([1, 2, 3])
-        self.translation = load_dataset("opus100", language_pair="de-nl", split="test").flatten()\
+        self.translation = load_dataset("opus100", "de-nl", split="test").flatten()\
             .rename_columns({"translation.de": "german", "translation.nl": "dutch"}).select([1, 2, 3])
 
     def test_translation(self):