Set up a parser for input arguments

setup_argparse.py

import argparse

# MAIN
parser = argparse.ArgumentParser(description='Input arguments for MLflow testing in general')
parser.add_argument('--run_name', type=str, 
                    help='Name of the run within the experiment family')
parser.add_argument('--exp_name', type=str,
                    help='Name of the family experiment')
parser.add_argument('--model', type=str,
                    help='Model to be used: naivebayes, logreg, randomforest')
parser.add_argument('--vectorizer', type=str,
                    help='Vectorizer to process text: countvectorizer or tfidf')


args = parser.parse_args()
run_name_ = args.run_name 
exp_name_ = args.exp_name 
model_ = args.model 
vectorizer_ = args.vectorizer