49 lines
1.4 KiB
Python
49 lines
1.4 KiB
Python
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)
|
|
if (args.rgb):
|
|
print(env.render(mode="rgb_array"))
|
|
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='OpenGL Visualizer', type=int, default=0)
|
|
parser.add_argument('--rgb',help='rgb_array gym rendering',type=int, 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()
|