from osim.env import ProstheticsEnv

env = ProstheticsEnv(seed=5)
env.change_model(model='3D', prosthetic=True, difficulty=2, seed=11)
observation = env.reset()
for i in range(300):
    observation, reward, done, info = env.step(env.action_space.sample(), project = False)
    if done:
        env.reset(seed=10)