From 301abc69554fa596d0fa1c3697ca7da200b95244 Mon Sep 17 00:00:00 2001
From: JulienPeloton <peloton@lal.in2p3.fr>
Date: Wed, 9 Aug 2023 12:41:20 +0200
Subject: [PATCH] Create folder if need be

---
 fink_spins/generate_data.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/fink_spins/generate_data.py b/fink_spins/generate_data.py
index f35f3cc..e3c4215 100644
--- 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"}