add simple script to test some of the pybullet gym environments (work-in-progress/experimental)

This commit is contained in:
Erwin Coumans
2017-09-08 15:25:16 -07:00
parent 67eb59901a
commit c144d9c045

View File

@@ -0,0 +1,45 @@
import os
import inspect
currentdir = os.path.dirname(os.path.abspath(inspect.getfile(inspect.currentframe())))
parentdir = os.path.dirname(os.path.dirname(currentdir))
os.sys.path.insert(0,parentdir)
import pybullet_envs
import gym
import argparse
def test(args):
env = gym.make(args.env)
env.env.configure(args)
if (args.render):
env.render(mode="human")
env.reset()
print("action space:")
sample = env.action_space.sample()
action = sample*0.0
print("action=")
print(action)
for i in range(args.steps):
obs,rewards,done,_ =env.step(action)
print("obs=")
print(obs)
print("rewards")
print (rewards)
print ("done")
print(done)
def main():
import argparse
parser = argparse.ArgumentParser(formatter_class=argparse.ArgumentDefaultsHelpFormatter)
parser.add_argument('--env', help='environment ID', default='AntBulletEnv-v0')
parser.add_argument('--seed', help='RNG seed', type=int, default=0)
parser.add_argument('--render', help='Render in OpenGL Window', type=str, default=0)
parser.add_argument('--steps', help='Number of steps', type=int, default=1)
parser.add_argument('--bla',help='bla',type=int, default=42)
args = parser.parse_args()
test(args)
if __name__ == '__main__':
main()