--- a +++ b/examples/under-construction/arm2d.py @@ -0,0 +1,15 @@ +import os +from osim.env import Arm2DEnv +import pprint +import numpy as np + +env = Arm2DEnv() +if __name__ == '__main__': + observation = env.reset() + env.osim_model.list_elements() + for i in range(3000): + action = env.action_space.sample() + observation, reward, done, info = env.step(action) + print(observation) + if done: + env.reset()