Jelajahi Sumber

factorize seeding in manual_control

saleml 2 tahun lalu
induk
melakukan
37aac4dcca
1 mengubah file dengan 6 tambahan dan 4 penghapusan
  1. 6 4
      manual_control.py

+ 6 - 4
manual_control.py

@@ -8,17 +8,16 @@ import gym_minigrid
 from gym_minigrid.wrappers import *
 from gym_minigrid.window import Window
 
+
 def redraw(img):
     if not args.agent_view:
         img = env.render(tile_size=args.tile_size)
 
     window.show_img(img)
 
+
 def reset():
-    if args.seed == -1:
-        seed = None
-    else:
-        seed = args.seed
+    seed = None if args.seed == -1 else args.seed
     obs = env.reset(seed=seed)
 
     if hasattr(env, 'mission'):
@@ -27,6 +26,7 @@ def reset():
 
     redraw(obs)
 
+
 def step(action):
     obs, reward, done, info = env.step(action)
     print('step=%s, reward=%.2f' % (env.step_count, reward))
@@ -37,6 +37,7 @@ def step(action):
     else:
         redraw(obs)
 
+
 def key_handler(event):
     print('pressed', event.key)
 
@@ -73,6 +74,7 @@ def key_handler(event):
         step(env.actions.done)
         return
 
+
 parser = argparse.ArgumentParser()
 parser.add_argument(
     "--env",