Fix tests/test_dataset_common.py (#221)

* fix-tests * fix-tests * fix-tests * fix-tests * style-test-dataset-common
huggingface · May 29, 2020 · 5778a27 · 5778a27
1 parent 32c3715
commit 5778a27
Showing 1 changed file with 2 additions and 8 deletions.
diff --git a/tests/test_dataset_common.py b/tests/test_dataset_common.py
@@ -168,12 +168,9 @@ def test_load_real_dataset(self, dataset_name):
         with tempfile.TemporaryDirectory() as temp_data_dir:
             download_config = DownloadConfig()
             download_config.download_mode = GenerateMode.FORCE_REDOWNLOAD
-            download_and_prepare_kwargs = {"download_config": download_config}
 
             dataset = load_dataset(
-                "./datasets/" + dataset_name,
-                data_dir=temp_data_dir,
-                download_and_prepare_kwargs=download_and_prepare_kwargs,
+                "./datasets/" + dataset_name, data_dir=temp_data_dir, download_config=download_config
             )
             for split in dataset.keys():
                 self.assertTrue(len(dataset[split]) > 0)
@@ -236,10 +233,7 @@ def test_load_real_dataset(self, dataset_name):
         with tempfile.TemporaryDirectory() as temp_data_dir:
             download_config = DownloadConfig()
             download_config.download_mode = GenerateMode.FORCE_REDOWNLOAD
-            download_and_prepare_kwargs = {"download_config": download_config}
 
-            dataset = load_dataset(
-                dataset_name, data_dir=temp_data_dir, download_and_prepare_kwargs=download_and_prepare_kwargs
-            )
+            dataset = load_dataset(dataset_name, data_dir=temp_data_dir, download_config=download_config)
             for split in dataset.keys():
                 self.assertTrue(len(dataset[split]) > 0)