File size: 647 Bytes
4b714e2
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
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]