combine download and extract_archive

pytorch · Jul 22, 2019 · 81e5a31 · 81e5a31
1 parent 58e3bac
commit 81e5a31
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/torchtext/datasets/text_classification.py b/torchtext/datasets/text_classification.py
@@ -9,7 +9,7 @@
 from tqdm import tqdm
 
 
-def download(url, raw_folder, dataset_name):
+def download_extract_archive(url, raw_folder, dataset_name):
     """Download the dataset if it doesn't exist in processed_folder already."""
 
     train_csv_path = os.path.join(raw_folder,
@@ -149,7 +149,7 @@ def __init__(self, url, root='.data',
                                              'processed')
         filepath = os.path.join(self.processed_folder, self.dataset_name + '.train')
         if not os.path.isfile(filepath):
-            download(self.url, self.raw_folder, self.dataset_name)
+            download_extract_archive(self.url, self.raw_folder, self.dataset_name)
             _preprocess(self.raw_folder, self.processed_folder, self.dataset_name)
         with open(filepath) as src_data:
             self.train_examples = _load_text_classification_data(src_data, self.fields, ngrams)