浏览代码

Fixed bug in manual_control.py reset

Maxime Chevalier-Boisvert 5 年之前
父节点
当前提交
ac41d86efa
共有 1 个文件被更改,包括 2 次插入2 次删除
  1. 2 2
      manual_control.py

+ 2 - 2
manual_control.py

@@ -33,8 +33,8 @@ def step(action):
     if done:
         print('done!')
         reset()
-
-    redraw(obs)
+    else:
+        redraw(obs)
 
 def key_handler(event):
     print('pressed', event.key)