57544b1ff9f97d4da9f64d25c8ea5a3d8d247ffc,rllib/examples/hierarchical_training.py,,,#,35

Before Change

        )

    // Error if stop-reward not reached.
    if results.trials[0].last_result["episode_reward_mean"] < \
            args.stop_reward:
        raise ValueError("`stop-reward` of {} not reached!".format(
            args.stop_reward))
    print("ok")

After Change

parser.add_argument("--flat", action="store_true")
parser.add_argument("--as-test", action="store_true")
parser.add_argument("--torch", action="store_true")
parser.add_argument("--stop-iters", type=int, default=200)
parser.add_argument("--stop-reward", type=float, default=0.0)
parser.add_argument("--stop-timesteps", type=int, default=100000)

logger = logging.getLogger(__name__)

Instances