import opensim as osim
from osim.http.client import Client
from osim.env import ProstheticsEnv
import numpy as np
import argparse
from pprint import pprint

# Settings
remote_base = 'http://grader.crowdai.org:1729'

# Command line parameters
parser = argparse.ArgumentParser(description='Submit the result to crowdAI')
parser.add_argument('--token', dest='token', action='store', required=True)
args = parser.parse_args()

client = Client(remote_base)

# Create environment
observation = client.env_create(args.token, env_id="ProstheticsEnv")
env = ProstheticsEnv()

# Run a single step
#
# The grader runs 3 simulations of at most 1000 steps each. We stop after the last one
while True:
    print(observation)
    [observation, reward, done, info] = client.env_step(env.action_space.sample().tolist())
    if done:
        observation = client.env_reset()
        if not observation:
            break

client.submit()

@@ -126,34 +126,8 @@
 arse
-%0Afrom pprint import pprint
 %0A%0A# 
@@ -558,10 +558,8 @@
 tep%0A
-#%0A
 # Th
@@ -868,16 +868,28 @@
   break%0A
+            
 %0Aclient.

Update submit.py