Function quit_if_window_closed

3-atari/env.py:123–133

Exit cleanly when the user clicks the window's X. No-op on headless runs (no pygame display initialized).

(env)

def quit_if_window_closed(env):
    """Exit cleanly when the user clicks the window's X.

    No-op on headless runs (no pygame display initialized).
    """
    if not pygame.display.get_init():
        return
    for event in pygame.event.get():
        if event.type == pygame.QUIT:
            env.close()
            sys.exit()


def run_test_loop(env, get_action):