Function run_test_loop

3-atari/env.py:136–148

Replay episodes forever using the supplied action picker.

(env, get_action)

def run_test_loop(env, get_action):
    """Replay episodes forever using the supplied action picker."""
    while True:
        obs, _ = env.reset()
        done = False
        score = 0.0
        while not done:
            quit_if_window_closed(env)
            action = get_action(np.asarray(obs))
            obs, reward, terminated, truncated, _ = env.step(action)
            done = terminated or truncated
            score += reward
        print(f"test score: {score}")