Merge pull request #449 from JohnSnowLabs/fix/XSum-test

hot-fix/XSum-test
JohnSnowLabs · May 24, 2023 · eff51e9 · eff51e9
2 parents 3b4e68b + 9178d89
commit eff51e9
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/nlptest/datahandler/datasource.py b/nlptest/datahandler/datasource.py
@@ -131,7 +131,8 @@ def _load_dataset(cls, dataset_name: str):
             'NQ-open-test': script_dir[:-7]+'/NQ-open/test.jsonl',
             'NQ-open': script_dir[:-7]+'/NQ-open/combined.jsonl',
             'NQ-open-test-tiny': script_dir[:-7]+'/NQ-open/test-tiny.jsonl',
-            'XSum-test-tiny' : script_dir[:-7]+'/Xsum/XSum-test-tiny.jsonl'
+            'XSum-test-tiny' : script_dir[:-7]+'/Xsum/XSum-test-tiny.jsonl',
+            'XSum-test' : script_dir[:-7]+'/Xsum/XSum-test.jsonl',
         }
         return datasets_info[dataset_name]