import gym
from prl.agents import RandomAgent
from prl.environments.environments import Environment
from prl.utils import time_logger

gym_env = gym.make("CartPole-v0")
env = Environment(gym_env, expected_episode_length=200)
agent = RandomAgent()
agent.train(env, 100)
print(time_logger)