def parse_arg()

in src/mlmax/monitoring.py [0:0]

• 19 lines of code
• 1 McCabe index (conditional complexity)

def parse_arg():
    parser = argparse.ArgumentParser()
    parser.add_argument("--mode", type=str, default="train")
    parser.add_argument("--data_dir", type=str, default="/opt/ml/processing")
    parser.add_argument(
        "--train_input", type=str, default="train_input/census-income.csv"
    )
    parser.add_argument(
        "--infer_input", type=str, default="infer_input/census-income.csv"
    )
    parser.add_argument(
        "--infer_baseline",
        type=str,
        default="profiling/baseline/train_features_baseline.csv",
    )
    parser.add_argument("--train_test_split_ratio", type=float, default=0.3)
    args, _ = parser.parse_known_args()
    logger.info(f"Received arguments {args}")
    return args