22 lines
535 B
Python
22 lines
535 B
Python
|
#!/usr/bin/env python3
|
||
|
|
import argparse
|
||
|
|
import gym
|
||
|
|
|
||
|
|
|
||
|
|
parser = argparse.ArgumentParser(description='Renders a Gym environment for quick inspection.')
|
||
|
|
parser.add_argument('env_id', type=str, help='the ID of the environment to be rendered (e.g. HalfCheetah-v1')
|
||
|
|
parser.add_argument('--step', type=int, default=1)
|
||
|
|
args = parser.parse_args()
|
||
|
|
|
||
|
|
env = gym.make(args.env_id)
|
||
|
|
env.reset()
|
||
|
|
|
||
|
|
step = 0
|
||
|
|
while True:
|
||
|
|
if args.step:
|
||
|
|
env.step(env.action_space.sample())
|
||
|
|
env.render()
|
||
|
|
if step % 10 == 0:
|
||
|
|
env.reset()
|
||
|
|
step += 1
|