Improve LightningModule hook tests (#7944)

Lightning-AI · Jun 14, 2021 · 03e7bdf · 03e7bdf
1 parent 3a0ed02
commit 03e7bdf
Show file tree

Hide file tree

Showing 2 changed files with 129 additions and 240 deletions.
diff --git a/tests/helpers/datasets.py b/tests/helpers/datasets.py
@@ -105,7 +105,7 @@ def prepare_data(self, download: bool = True):
             raise RuntimeError('Dataset not found.')
 
     def _download(self, data_folder: str) -> None:
-        os.makedirs(data_folder)
+        os.makedirs(data_folder, exist_ok=True)
         for url in self.RESOURCES:
             logging.info(f'Downloading {url}')
             fpath = os.path.join(data_folder, os.path.basename(url))