From 60c954c8c116e3b9a98f65ed02b9c19bcaedb550 Mon Sep 17 00:00:00 2001
From: Dominik Roth <dominik.roth.dev@gmail.com>
Date: Sat, 25 Jun 2022 21:47:21 +0200
Subject: [PATCH] LunarLanderContinuous-v2 is our new default test-env

---
 test.py | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

diff --git a/test.py b/test.py
index cb32cfb..7794d60 100755
--- a/test.py
+++ b/test.py
@@ -17,7 +17,7 @@ import columbus
 root_path = '.'
 
 
-def main(env_name='ColumbusCandyland_Aux10-v0', timesteps=5000, showRes=False, saveModel=True, n_eval_episodes=8):
+def main(env_name='ColumbusCandyland_Aux10-v0', timesteps=50000, showRes=False, saveModel=True, n_eval_episodes=16):
     env = gym.make(env_name)
     test_sde = False
     ppo = PPO(
@@ -87,12 +87,12 @@ def main(env_name='ColumbusCandyland_Aux10-v0', timesteps=5000, showRes=False, s
     #    learning_rate=0.001 # 0.015
     # )
 
-    print('PPO:')
-    testModel(ppo, timesteps, showRes,
-              saveModel, n_eval_episodes)
     print('TRL_PG:')
     testModel(trl_pg, timesteps, showRes,
               saveModel, n_eval_episodes)
+    print('PPO:')
+    testModel(ppo, timesteps, showRes,
+              saveModel, n_eval_episodes)
 
 
 def testModel(model, timesteps, showRes=False, saveModel=False, n_eval_episodes=16):
@@ -131,4 +131,4 @@ def testModel(model, timesteps, showRes=False, saveModel=False, n_eval_episodes=
 
 
 if __name__ == '__main__':
-    main('CartPole-v1')
+    main('LunarLanderContinuous-v2')