57544b1ff9f97d4da9f64d25c8ea5a3d8d247ffc,rllib/examples/nested_action_spaces.py,,,#,12

Before Change

    for _ in range(100):
        results = trainer.train()
        print(results)
        if results["episode_reward_mean"] > args.stop:
            sys.exit(0)  // Learnt, exit gracefully.
    sys.exit(1)  // Done, but did not learn, exit with error.

After Change

    if args.as_test:
        check_learning_achieved(results, args.stop_reward)

    ray.shutdown()

Instances