Testing OpenAI's Gym

test_gym.py

import gym
import time
from gym import envs

print(envs.registry.all())

env = gym.make('CartPole-v0')
env.reset()
t = 0

while True:
    time.sleep(0.1)
    env.render()
    obs, r, done, inf = env.step(env.action_space.sample())
    if done:
        print("Resetting. Time:", t)
        env.reset()
        t = 0
    else:
        t += 1