diff --git a/mlair/data_handler/default_data_handler.py b/mlair/data_handler/default_data_handler.py index 73b6b53d3be657d5b14796e26c38ac1364d51112..f70f454b2cc2cd9113565c4403969d08fa47072a 100644 --- a/mlair/data_handler/default_data_handler.py +++ b/mlair/data_handler/default_data_handler.py @@ -80,7 +80,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)