diff --git a/mlair/data_handler/default_data_handler.py b/mlair/data_handler/default_data_handler.py index 11461ad77c3e910a897a9a1be48aef7cef45480a..181e6604d405163c2f77f84a0b176f620b913f29 100644 --- a/mlair/data_handler/default_data_handler.py +++ b/mlair/data_handler/default_data_handler.py @@ -79,7 +79,7 @@ class DefaultDataHandler(AbstractDataHandler): def _cleanup(self): directory = os.path.dirname(self._save_file) if os.path.exists(directory) is False: - os.makedirs(directory) + os.makedirs(directory, exist_ok=True) if os.path.exists(self._save_file): shutil.rmtree(self._save_file, ignore_errors=True)