From 85cc83a930010d4bb31f85ced9bd3ab5737905cf Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Ga=C3=A9tan=20Ramet?= Date: Tue, 23 Apr 2024 14:18:17 +0200 Subject: [PATCH] added output folder creation --- src/pipeline/preprocess/preprocessing.py | 1 + src/pipeline/train/training.py | 1 + 2 files changed, 2 insertions(+) diff --git a/src/pipeline/preprocess/preprocessing.py b/src/pipeline/preprocess/preprocessing.py index 264eec6..abcd121 100644 --- a/src/pipeline/preprocess/preprocessing.py +++ b/src/pipeline/preprocess/preprocessing.py @@ -7,6 +7,7 @@ def main(input_folder: Path = typer.Option(...), output_folder: Path = typer.Option(...)) -> None: """Main function.""" + output_folder.mkdir(exist_ok=True, parents=True) if __name__ == "__main__": diff --git a/src/pipeline/train/training.py b/src/pipeline/train/training.py index 94f3ff3..0c14b0b 100644 --- a/src/pipeline/train/training.py +++ b/src/pipeline/train/training.py @@ -7,6 +7,7 @@ def main(preprocess_folder: Path = typer.Option(...), output_folder: Path = typer.Option(...)) -> None: """Main function.""" + output_folder.mkdir(exist_ok=True, parents=True) if __name__ == "__main__":