def init():
tools = Tools()
for i in range(200):
observation, reward, done, info = env.step(
tools.active_one_muscle("iliopsoas", "r", 1))
if i == 20:
state_desc = env.get_state_desc()
print(type(state_desc["body_pos"]["toes_r"][0:2]))
print(state_desc["body_pos"]["talus_l"][0:2])
print(state_desc["body_pos"]["talus_r"][0:2])
print(state_desc["misc"]["mass_center_pos"])
print(state_desc["body_pos_rot"])
input("Press Enter to continue...")
print(reward)
