added output folder creation

VisiumCH · Apr 23, 2024 · 85cc83a · 85cc83a
1 parent f74825f
commit 85cc83a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git 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
@@ -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__":