def add_args()

in src/entrypoint/train.py [0:0]

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

def add_args(parser: ArgumentParser):
    """Configure hyperparameters captured by this entrypoint script."""
    parser.add_argument(
        "--algo",
        type=str,
        help="Estimator class",
        default=os.environ.get("SM_HP_ALGO", "gluonts.model.deepar.DeepAREstimator"),
    )
    parser.add_argument(
        "--dataset",
        type=str,
        help="When s3_dataset channel not specified, fallback to this public dataset.",
        default=os.environ.get("SM_HP_DATASET", ""),
    )
    parser.add_argument(
        "--y_transform",
        type=str,
        help="Transformation to apply on target variable.",
        default="noop",
        choices=["noop", "log1p"],
    )
    parser.add_argument(
        "--num_samples",
        type=int,
        help="Number of samples for backtesting.",
        default=os.environ.get("SM_HP_NUM_SAMPLES", 1000),
    )
    parser.add_argument(
        "--quantiles",
        help="Quantiles for backtesting",
        default=os.environ.get("SM_HP_QUANTILES", [0.1, 0.2, 0.3, 0.4, 0.5, 0.6, 0.7, 0.8, 0.9]),
    )
    parser.add_argument(
        "--plot_transparent",
        type=int,
        help="Whether plots use transparent background.",
        default=os.environ.get("SM_HP_PLOT_TRANSPARENT", 0),
    )
    parser.add_argument("--stop_before", type=str, help="For debug/dev/test", default="", choices=["", "train", "eval"])