Browse Source

Minor fix to speed up tests

Maxime Chevalier-Boisvert 6 years ago
parent
commit
ed9c2e7f75
1 changed files with 1 additions and 0 deletions
  1. 1 0
      run_tests.py

+ 1 - 0
run_tests.py

@@ -21,6 +21,7 @@ for envName in env_list:
 
     # Load the gym environment
     env = gym.make(envName)
+    env.max_steps = min(env.max_steps, 200)
     env.reset()
     env.render('rgb_array')