diff --git a/femnist.py b/femnist.py index 278e103..8f8df79 100644 --- a/femnist.py +++ b/femnist.py @@ -50,8 +50,8 @@ def download(self): if self._check_exists(): return - utils.makedir_exist_ok(self.raw_folder) - utils.makedir_exist_ok(self.processed_folder) + os.makedirs(self.raw_folder, exist_ok=True) + os.makedirs(self.processed_folder, exist_ok=True) # download files for url, md5 in self.resources: @@ -61,4 +61,4 @@ def download(self): # process and save as torch files print('Processing...') shutil.move(os.path.join(self.raw_folder, self.training_file), self.processed_folder) - shutil.move(os.path.join(self.raw_folder, self.test_file), self.processed_folder) + shutil.move(os.path.join(self.raw_folder, self.test_file), self.processed_folder) \ No newline at end of file