add simple script to test some of the pybullet gym environments (work-in-progress/experimental)
This commit is contained in:
45
examples/pybullet/gym/pybullet_envs/examples/testEnv.py
Normal file
45
examples/pybullet/gym/pybullet_envs/examples/testEnv.py
Normal 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()
|
||||
Reference in New Issue
Block a user