def get_parser(): parser = argparse.ArgumentParser() parser.add_argument('--fpath', type=str) parser.add_argument('--len', '-l', type=int, default=0) parser.add_argument('--norender', '-nr', action='store_true') parser.add_argument('--itr', '-i', type=int, default=-1) parser.add_argument('--deterministic', '-d', action='store_true') parser.add_argument('--episode_ids', '-id', type=str, default="0") parser.add_argument('--bests', type=str2bool, default=None) parser.add_argument('--fixed_test_set', '-ts', type=str, default=None) parser.add_argument('--load_env', action='store_true') parser.add_argument('--use_test_env', action='store_true') parser.add_argument('--record', type=str2bool, default=False) parser.add_argument('--recording_path', type=str, default=None) EnvironmentArgsHandler.set_parser_arguments(parser) StudentArgsHandler.set_parser_arguments(parser) return parser
'env': 'parametric-continuous-parkour-v0', "walker_type": ["old_classic_bipedal"], "movable_creepers": None, }, # { # 'env':'parametric-continuous-flat-parkour-v0', # "walker_type": ["classic_bipedal", "fish"], # "motors_torque": 80, # "water_level": 4, # "dummy_param": [0, 1] # } ] parser = argparse.ArgumentParser() parser.add_argument('--seed', '-s', type=int, default=0) StudentArgsHandler.set_parser_arguments(parser) EnvironmentArgsHandler.set_parser_arguments(parser) TeacherArgsHandler.set_parser_arguments(parser) for _env in envs: _env["teacher"] = 'Random' _env["nb_test_episodes"] = 1 _env["seed"] = 43 current_args_dict = deepcopy(_env) for walker_type in _env["walker_type"]: print("##### Benchmarking {0} with {1} body #####".format( _env['env'], walker_type)) current_args_dict["walker_type"] = walker_type args_str = dict_to_args_str(current_args_dict) args = parser.parse_args(args_str) for j in range(2):