Create folder if need be

astrolabsoftware · Aug 9, 2023 · 301abc6 · 301abc6
1 parent e37700d
commit 301abc6
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/fink_spins/generate_data.py b/fink_spins/generate_data.py
@@ -18,6 +18,7 @@
 import logging
 import io
 import sys
+import os
 
 import numpy as np
 import pandas as pd
@@ -115,6 +116,7 @@ def format_ssoft(data: pd.DataFrame, filters: dict, flavor: str) -> pd.DataFrame
         help="If specified, print on screen information about the processing.")
     args = parser.parse_args(None)
 
+    os.makedirs(args.output_dir, exist_ok=True)
     # ZTF filters 1: g, 2: r
     filters = {"1": "g", "2": "r"}