10 wait until a key is pressed

def wait():
    while (pygame.event.wait().type != pygame.KEYDOWN): pass
    return