57544b1ff9f97d4da9f64d25c8ea5a3d8d247ffc,rllib/examples/twostep_game.py,,,#,21

Before Change

    ray.init(num_cpus=args.num_cpus or None)
    tune.run(
        args.run,
        stop={
            "timesteps_total": args.stop,
        },
        config=dict(config, **{
            "env": "grouped_twostep" if group else TwoStepGame,
        }),

After Change

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

    ray.shutdown()

Instances