Store true

lstmcpipe/hiperta/hiperta_r0_to_dl1lstchain.py

@@ -5,7 +5,6 @@
 
 import os
 import argparse
-from distutils.util import strtobool
 from lstmcpipe.hiperta.reorganize_dl1hiperta300_to_dl1lstchain060 import (
     main as reorganize_dl1,
 )
@@ -48,19 +47,17 @@ def main():
     parser.add_argument(
         "--keep_file",
         "-k",
-        type=lambda x: bool(strtobool(x)),
+        action='store_true',
         dest="keep_file",
         help="Keep output of hiperta. Set by default to False",
-        default=False,
     )
 
     parser.add_argument(
         "--debug_mode",
         "-d",
-        type=lambda x: bool(strtobool(x)),
+        action='store_true',
         dest="debug_mode",
         help="Activate debug mode (add cleaned mask in the output hdf5). Set by default to False",
-        default=False,
     )
     args = parser.parse_args()
 

lstmcpipe/scripts/script_batch_filelist_rta.py

@@ -2,7 +2,6 @@
 
 import argparse
 import subprocess
-from distutils.util import strtobool
 import os
 
 
@@ -39,18 +38,16 @@ def main():
     parser.add_argument(
         "--keep_file",
         "-k",
-        type=lambda x: bool(strtobool(x)),
+        action='store_true',
         dest="keep_file",
         help="Keep output of hiperta. Set by default to False",
-        default=False,
     )
     parser.add_argument(
         "--debug_mode",
         "-d",
-        type=lambda x: bool(strtobool(x)),
+        action='store_true',
         dest="debug_mode",
         help="Activate debug mode (add cleaned mask in the output hdf5). Set by default to False",
-        default=False,
     )
     args = parser.parse_args()
 
@@ -66,11 +63,13 @@ def main():
                 "lstmcpipe_hiperta_r0_to_dl1lstchain",
                 f"--infile={file}",
                 f"--outdir={args.output_dir}",
-                f"--keep_file={args.keep_file}",
-                f"--debug_mode={args.debug_mode}",
+                f"--config={args.config_file}",
             ]
-            if args.config_file:
-                cmd.append(f"--config={args.config_file}")
+
+            if args.keep_file:
+                cmd.append(f"--keep_file")
+            if args.debug_mode:
+                cmd.append(f"--debug_mode")
 
             subprocess.run(cmd)
 

lstmcpipe/stages/mc_r0_to_dl1.py

@@ -129,6 +129,7 @@ def r0_to_dl1(
     workflow_kind="lstchain",
     keep_rta_file=False,
     n_jobs_parallel=20,
+    debug_mode=False,
 ):
     """
     R0 to DL1 MC onsite conversion.
@@ -170,12 +171,18 @@ def r0_to_dl1(
         ! NOTE : train_pipe AND dl1_to_dl2 **MUST** be run with the same environment.
     workflow_kind: str
         One of the supported pipelines. Defines the command to be run on r0 files
-    keep_rta_file : bool
-        Argument to be passed to the hiperta_r0_to_dl1lstchain script, which runs the hiperta_r0_dl1 and
-        re-organiser stage
     n_jobs_parallel: int
         Number of jobs to be run at the same time per array.
 
+    # HIPERTA ARGUMENTS
+    keep_rta_file : bool
+        Flag to indicate whether to keep (keep_rta_file = True) or remove (keep_rta_file = Flase ) the
+        `dl1v06_reorganized_*.h5` output file (hiperta_r0_dl1 and re-organiser stages).
+        Argument to be passed to the hiperta_r0_to_dl1lstchain script.
+    debug_mode : bool
+        Flag to activate debug_mode. Only compatible with `hiperta` workflow kind, i.e.,
+        HiPeRTA functionality. DEFAULT=False.
+
     Returns
     -------
     jobid2log : dict
@@ -208,8 +215,12 @@ def r0_to_dl1(
     elif workflow_kind == "hiperta":
         base_cmd = (
             f"{source_environment} lstmcpipe_rta_core_r0_dl1 -c {config_file} "
-            f"--debug_mode False -k {keep_rta_file}"
         )
+        if keep_rta_file:
+            base_cmd += " --keep_rta_file"
+        if debug_mode:
+            base_cmd += " --debug_mode"
+
         jobtype_id = "RTA"
     else:
         log.critical("Please, selected an allowed workflow kind.")