Function run_test_loop

4-atari-hard/env.py:228–240

Replay episodes forever using the supplied action picker (single env).

(env, get_action)

def run_test_loop(env, get_action):
    """Replay episodes forever using the supplied action picker (single env)."""
    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}")