def play_game(env):
    """
    Play a game in the given environment and return the collected
    observations
    """
    observations=[]
    observation=env.reset()
    done=False

    while not done:
        act=env.act(observation)
        observation, rwd, done, _ = env.step(act)
        observations.append(observation)
        #env.render()
    env.close()

    return observations