Переглянути джерело

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)