Slightly refactored enjoy

bharatr21 · May 31, 2018 · d8a8dda · d8a8dda
1 parent 4bc0dd0
commit d8a8dda
Showing 1 changed file with 7 additions and 7 deletions.
diff --git a/rl/scripts/enjoy.py b/rl/scripts/enjoy.py
@@ -49,22 +49,22 @@
 
 # Run the agent
 
-obs = env.reset()
-done = False
-print("Mission: {}".format(obs["mission"]))
+done = True
 
 while True:
     time.sleep(args.pause)
     renderer = env.render("human")
 
     if done:
-        print("Reward:", reward)
         obs = env.reset()
         print("Mission: {}".format(obs["mission"]))
 
-    if renderer.window is None:
-        break
-
     action = agent.get_action(obs)
     obs, reward, done, _ = env.step(action)
     agent.analyze_feedback(reward, done)
+
+    if done:
+        print("Reward:", reward)
+
+    if renderer.window is None:
+        break