🧪 Fix bug in test_datasets.py

hezarai · Jun 9, 2024 · badb361 · badb361
1 parent f506b36
commit badb361
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/test_datasets.py b/tests/test_datasets.py
@@ -78,6 +78,7 @@ def test_load_dataset(task):
     test_dataset = Dataset.load(
         path,
         split="test",
+        preprocessor=preprocessor,
         **DATASETS_MAPPING[task]
     )
     assert isinstance(test_dataset, Dataset), INVALID_DATASET_TYPE.format(type(test_dataset))