def METHOD_NAME(name):
    env = gym.make(name)
    _, _, mask = env.reset()
    done = False
    while not done:
        aa = np.where(mask > 0.0)[0]
        action_idx = np.random.choice(aa, 1)[0]
        (_, _, mask), _, done, _ = env.step(action_idx)