459038549c5f9fb0919ae0b0b90077ca0a035c68,examples/ale/train_a2c_ale.py,,main,#,51

Before Change

            args.eval_n_runs, eval_stats["mean"], eval_stats["median"],
            eval_stats["stdev"]))
    else:
        experiments.train_agent_batch_with_evaluation(
            agent=agent,
            env=make_batch_env(test=False),
            eval_env=make_batch_env(test=True),
            steps=args.steps,
            eval_n_steps=None,
            eval_n_episodes=args.eval_n_runs,
            eval_interval=args.eval_interval,
            outdir=args.outdir,
            save_best_so_far_agent=False,
            log_interval=1000,
        )


if __name__ == "__main__":
    main()

After Change

            eval_stats["stdev"]))
        env.close()
    else:
        env = make_batch_env(test=False)
        eval_env = make_batch_env(test=True)
        experiments.train_agent_batch_with_evaluation(
            agent=agent,
            env=env,
            eval_env=eval_env,
            steps=args.steps,
            eval_n_steps=None,
            eval_n_episodes=args.eval_n_runs,
            eval_interval=args.eval_interval,
            outdir=args.outdir,
            save_best_so_far_agent=False,
            log_interval=1000,
        )
        env.close()
        eval_env.close()


if __name__ == "__main__":
    main()

Instances