68ec284dd57ba00230c6c6199cbb71d53e0696fc,scripts/visualize.py,,,#,11

Before Change

        obs = env.reset()

    time.sleep(args.pause)
    renderer = env.render()
    if args.gif:
        frames.append(numpy.moveaxis(env.render("rgb_array"), 2, 0))

    action = agent.get_action(obs)
    obs, reward, done, _ = env.step(action)
    agent.analyze_feedback(reward, done)

    if renderer.window is None:
        if args.gif:
            print("Saving gif... ", end="")
            write_gif(numpy.array(frames), args.gif+".gif", fps=1/args.pause)
            print("Done.")
        break

After Change

done = True

for episode in range(args.episodes):
    obs = env.reset()

    while True:
        time.sleep(args.pause)
        env.render("human")
        if args.gif:
            frames.append(numpy.moveaxis(env.render("rgb_array"), 2, 0))

        action = agent.get_action(obs)
        obs, reward, done, _ = env.step(action)
        agent.analyze_feedback(reward, done)

        if done:
            break

if args.gif:
    print("Saving gif... ", end="")
    write_gif(numpy.array(frames), args.gif+".gif", fps=1/args.pause)

Instances