input_args.py

import argparse

from apple.utils import str2bool


def apple_parse_args(args=None):
    parser = argparse.ArgumentParser()

    parser.add_argument("--c", type=float, default=0.25, required=False)
    parser.add_argument("--start_x", type=float, default=0.0, required=False)
    parser.add_argument("--goal_x", type=float, default=10.0, required=False)
    parser.add_argument("--time_limit", type=int, default=100.0, required=False)

    parser.add_argument("--lr", type=float, default=1e-3, required=False)
    parser.add_argument("--log_to_wandb", type=str2bool, default=True, required=False)

    return parser.parse_known_args(args=args)[0]