Simple Taxi Domain example using OpenAI Gym

taxi_simple.py

import gym
from gym.wrappers import Monitor

def main():
    GAME    = "Taxi-v2"
    env     = gym.make(GAME)
    n_state = env.observation_space.n
    n_action = env.action_space.n
    env = Monitor(env, "taxi_simple", force=True)

    s = env.reset()
    steps = 10
    for step in range(steps):
        env.render()
        action = env.action_space.sample()
        s, r, done, info = env.step(action)

    # close environment and monitor
    env.close()

if __name__ == '__main__':
    main()