diff --git a/reinforcement_learning/multi_agent_training.py b/reinforcement_learning/multi_agent_training.py index 340a21fc60e4ae2b9a68c008b7ddc30ea19be120..2d513ed84ae6ecc392aca195851db60e400d0523 100755 --- a/reinforcement_learning/multi_agent_training.py +++ b/reinforcement_learning/multi_agent_training.py @@ -510,7 +510,7 @@ def eval_policy(env, tree_observation, policy, train_params, obs_params): if __name__ == "__main__": parser = ArgumentParser() parser.add_argument("-n", "--n_episodes", help="number of episodes to run", default=5000, type=int) - parser.add_argument("--n_agent_fixed", help="hold the number of agent fixed", default=False, type=bool) + parser.add_argument("--n_agent_fixed", help="hold the number of agent fixed", action='store_true') parser.add_argument("-t", "--training_env_config", help="training config id (eg 0 for Test_0)", default=1, type=int) parser.add_argument("-e", "--evaluation_env_config", help="evaluation config id (eg 0 for Test_0)", default=1,