Merge pull request #37 from PgLoLo/master

Download of dataset requires to hold archive in memory

torchaudio/datasets/vctk.py

@@ -181,9 +181,7 @@ class VCTK(data.Dataset):
         filename = url.rpartition('/')[2]
         file_path = os.path.join(self.root, self.raw_folder, filename)
         if not os.path.isfile(file_path):
-            data = urllib.request.urlopen(url)
-            with open(file_path, 'wb') as f:
-                f.write(data.read())
+            urllib.request.urlretrieve(url, file_path)
         if not os.path.exists(dset_abs_path):
             with tarfile.open(file_path) as zip_f:
                 zip_f.extractall(raw_abs_dir)

torchaudio/datasets/yesno.py

@@ -101,9 +101,7 @@ class YESNO(data.Dataset):
         filename = url.rpartition('/')[2]
         file_path = os.path.join(self.root, self.raw_folder, filename)
         if not os.path.isfile(file_path):
-            data = urllib.request.urlopen(url)
-            with open(file_path, 'wb') as f:
-                f.write(data.read())
+            urllib.request.urlretrieve(url, file_path)
         else:
             print("Tar file already downloaded")
         if not os.path.exists(dset_abs_path):