Commit 68293a60 authored by Benjamin Beyret's avatar Benjamin Beyret
Browse files

testDocker: add try to agent reset

parent ab40db8b
......@@ -50,9 +50,14 @@ def main():
for k in range(5):
env.reset(arenas_configurations=arena_config_in)
submitted_agent.reset(t=arena_config_in.arenas[0].t)
cumulated_reward = 0
print('Episode {} starting'.format(k))
try:
submitted_agent.reset(t=arena_config_in.arenas[0].t)
except Exception as e:
print('Agent reset failed during episode {}'.format(k))
raise e
try:
obs, reward, done, info = env.step([0, 0])
for i in range(arena_config_in.arenas[0].t):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment