Fix monitor start bug #120

erlerobot · Apr 12, 2018 · 1ed9399 · 1ed9399
1 parent ea8c198
commit 1ed9399
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/examples/gazebo_cartpole/gazebo_cartpole_v1.py b/examples/gazebo_cartpole/gazebo_cartpole_v1.py
@@ -147,7 +147,7 @@ def run_episode(env, policy_grad, value_grad, sess):
 outdir = '/tmp/gazebo_gym_experiments'
 plotter = LivePlot(outdir)
 import time
-#env.monitor.start('cartpole-hill/', force=True)
+env = gym.wrappers.Monitor(env, outdir, force=True)
 policy_grad = policy_gradient()
 value_grad = value_gradient()
 sess = tf.InteractiveSession()
@@ -169,5 +169,4 @@ def run_episode(env, policy_grad, value_grad, sess):
     reward = run_episode(env, policy_grad, value_grad, sess)
     t += reward
 print (t / 1000)
-#env.monitor.close()
 print ("END!")