Skip to content

Commit

Permalink
move argparse into main
Browse files Browse the repository at this point in the history
  • Loading branch information
rlopezcoto committed Jan 10, 2022
1 parent f4eee36 commit 5b848e7
Show file tree
Hide file tree
Showing 10 changed files with 23 additions and 29 deletions.
4 changes: 2 additions & 2 deletions lstchain/scripts/lstchain_add_source_dependent_parameters.py
Original file line number Diff line number Diff line change
Expand Up @@ -42,10 +42,10 @@
default=None
)

args = parser.parse_args()

def main():

args = parser.parse_args()

dl1_filename = os.path.abspath(args.input_file)

config = get_standard_config()
Expand Down
5 changes: 2 additions & 3 deletions lstchain/scripts/lstchain_check_dl1.py
Original file line number Diff line number Diff line change
Expand Up @@ -73,11 +73,10 @@
help='Run the script without plotting output'
)

args, unknown = parser.parse_known_args()


def main():

args, unknown = parser.parse_known_args()

logging.basicConfig(level=logging.INFO, format='%(message)s')
logger = logging.getLogger(__name__)

Expand Down
8 changes: 3 additions & 5 deletions lstchain/scripts/lstchain_dl1_muon_analysis.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,8 +3,7 @@
"""
Script to perform the analysis of muon events.
- Inputs are a DL1a data file (pixel information is needed) and a
calibration file
- Inputs are a DL1a data file (pixel information is needed) and a calibration file
- Output is a table with muon parameters (to be updated to a dataframe!)
Usage:
Expand Down Expand Up @@ -81,11 +80,10 @@
help="Maximum number of processed muon ring candidates",
)

args = parser.parse_args()


def main():

args = parser.parse_args()

print("input files: {}".format(args.input_file))
print("calib file: {}".format(args.calib_file))
print("output file: {}".format(args.output_file))
Expand Down
5 changes: 2 additions & 3 deletions lstchain/scripts/lstchain_mc_r0_to_dl1.py
Original file line number Diff line number Diff line change
Expand Up @@ -43,11 +43,10 @@
default=None
)

args = parser.parse_args()


def main():

args = parser.parse_args()

output_dir = args.output_dir.absolute()
output_dir.mkdir(exist_ok=True)
output_file = output_dir / r0_to_dl1_filename(args.input_file.name)
Expand Down
7 changes: 3 additions & 4 deletions lstchain/scripts/lstchain_mc_r0_to_dl2.py
Original file line number Diff line number Diff line change
Expand Up @@ -14,10 +14,9 @@
--path-models ./trained_models
"""

from ctapipe.utils import get_dataset_path
import argparse
import os
from ctapipe.utils import get_dataset_path
from distutils.util import strtobool
from lstchain.paths import r0_to_dl1_filename
from pathlib import Path
Expand Down Expand Up @@ -53,10 +52,10 @@
default=None
)

args = parser.parse_args()


def main():
args = parser.parse_args()

outdir = args.outdir.absolute()
dl1_file = outdir / r0_to_dl1_filename(args.datafile.name)

Expand Down
4 changes: 2 additions & 2 deletions lstchain/scripts/lstchain_mc_rfperformance.py
Original file line number Diff line number Diff line change
Expand Up @@ -72,10 +72,10 @@
default=None
)

args = parser.parse_args()


def main():
args = parser.parse_args()

custom_config = {}
if args.config_file is not None:
try:
Expand Down
4 changes: 2 additions & 2 deletions lstchain/scripts/lstchain_mc_sensitivity.py
Original file line number Diff line number Diff line change
Expand Up @@ -51,10 +51,10 @@
dest = 'output_path',
help = 'path where to save plot images')

args = parser.parse_args()


def main():
args = parser.parse_args()

ntelescopes_gamma = 1
n_bins_energy = 20 # Number of energy bins
obstime = 50 * 3600 * u.s
Expand Down
5 changes: 2 additions & 3 deletions lstchain/scripts/lstchain_mc_trainpipe.py
Original file line number Diff line number Diff line change
Expand Up @@ -50,10 +50,9 @@
default=None
)


args = parser.parse_args()

def main():
args = parser.parse_args()

#Train the models

config = {}
Expand Down
5 changes: 2 additions & 3 deletions lstchain/scripts/lstchain_merge_muon_files.py
Original file line number Diff line number Diff line change
Expand Up @@ -46,10 +46,9 @@
default='*.fits',
)

args = parser.parse_args()


def main():
args = parser.parse_args()

if args.run_number:
run = f'Run{args.run_number:05d}'
file_list = sorted(filter(
Expand Down
5 changes: 3 additions & 2 deletions lstchain/scripts/lstchain_tune_nsb.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,5 @@
#!/usr/bin/env python3

"""
Usage:
Expand Down Expand Up @@ -42,10 +44,9 @@
required=True,
)

args = parser.parse_args()


def main():
args = parser.parse_args()

if not args.config.is_file():
log.error('Config file does not exist or is not a file')
Expand Down

0 comments on commit 5b848e7

Please sign in to comment.