githubmirror/gym | Gitverse

bin/render.py

21 строка535 B

Matthias Plappert

New robotics environments (#912)

26 фев 2018, 19:35

26 фев 2018, 19:3578c416e

#!/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