459038549c5f9fb0919ae0b0b90077ca0a035c68,examples/gym/train_iqn_gym.py,,main,#,33

Before Change

            args.eval_n_runs, eval_stats["mean"], eval_stats["median"],
            eval_stats["stdev"]))
    else:
        experiments.train_agent_with_evaluation(
            agent=agent,
            env=env,
            steps=args.steps,
            eval_n_steps=None,
            eval_n_episodes=args.eval_n_runs,
            eval_interval=args.eval_interval,
            outdir=args.outdir,
            eval_env=eval_env,
            train_max_episode_len=timestep_limit,
        )


if __name__ == "__main__":
    main()

After Change

            eval_stats["stdev"]))
        eval_env.close()
    else:
        experiments.train_agent_with_evaluation(
            agent=agent,
            env=env,
            steps=args.steps,
            eval_n_steps=None,
            eval_n_episodes=args.eval_n_runs,
            eval_interval=args.eval_interval,
            outdir=args.outdir,
            eval_env=eval_env,
            train_max_episode_len=timestep_limit,
        )
        env.close()
        eval_env.close()


if __name__ == "__main__":
    main()

Instances